pytorch的CUDA:CUDA C ++流和状态

Posted

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了pytorch的CUDA:CUDA C ++流和状态相关的知识,希望对你有一定的参考价值。

我正在尝试遵循this tutorial,并使用CUDA后端进行简单的c ++扩展。我的CPU实现似乎正常。

我在查找示例和文档时遇到了麻烦(似乎事情一直在变化)。

特别是

  1. 我看到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如何提供此信息以及如何获取?

  2. state然后转换为state由于某些原因,不再定义cudaStream_t stream = THCState_getCurrentStream(state);?如何从THCState_getCurrentStream中获取stream

[我也曾尝试在state论坛上提问-到目前为止无济于事。

答案

已弃用(无文档!)看这里:pytorch

简而言之:使用https://github.com/pytorch/pytorch/pull/14500

以上是关于pytorch的CUDA:CUDA C ++流和状态的主要内容,如果未能解决你的问题,请参考以下文章

C语言编程中如何将一个文件中的信息转入到另一个文件

Linux c学习--从标准输入输出看流和缓冲区

C语言中的文本流与二进制流的区别

C++学习摘要之九:C++流和文件流

IO流在C++中的应用

使用 Delphi 6(C/C++ 也可以)创建视频文件(AVI 等)的库或组件?