pytorch的CUDA:CUDA C ++流和状态
Posted
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了pytorch的CUDA:CUDA C ++流和状态相关的知识,希望对你有一定的参考价值。
我正在尝试遵循this tutorial,并使用CUDA后端进行简单的c ++扩展。我的CPU实现似乎正常。
我在查找示例和文档时遇到了麻烦(似乎事情一直在变化)。
特别是
我看到pytorch cuda函数获取
THCState *state
参数-该参数来自何处?我如何同时获得功能的state
?例如,在tensor.cat
的cuda实现中:tensor.cat
但是,当从python调用
void THCTensor_(cat)(THCState *state, THCTensor *result, THCTensor *ta, THCTensor *tb, int dimension)
时,不提供任何tensor.cat()
参数,pytorch在“幕后”提供了该参数。 pytorch如何提供此信息以及如何获取?state
然后转换为state
由于某些原因,不再定义cudaStream_t stream = THCState_getCurrentStream(state);
?如何从THCState_getCurrentStream
中获取stream
?
[我也曾尝试在state
论坛上提问-到目前为止无济于事。
答案
已弃用(无文档!)看这里:pytorch
简而言之:使用https://github.com/pytorch/pytorch/pull/14500
以上是关于pytorch的CUDA:CUDA C ++流和状态的主要内容,如果未能解决你的问题,请参考以下文章