代码之家  ›  专栏  ›  技术社区  ›  Shai

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

  •  0
  • Shai  · 技术社区  · 7 年前

    我正在努力跟上 this tutorial

    明确地

    1. THCState *state 论点——这个论点来自哪里?我怎样才能得到一份工作 state
      例如,在cuda中实现 tensor.cat :

      void THCTensor_(cat)(THCState *state, THCTensor *result, THCTensor *ta, THCTensor *tb, int dimension)
      

      tensor.cat() python中没有提供任何

    2. 状态 然后转换为 cudaStream_t stream = THCState_getCurrentStream(state);
      出于某种原因, THCState_getCurrentStream 不再有定义了吗?我怎样才能拿到票 stream 从我的 状态 ?

    我还试着继续问 pytorch 论坛-到目前为止没有结果。

    0 回复  |  直到 7 年前
        1
  •  2
  •   yfi    6 年前

    它已弃用(没有文档!) 请看这里: https://github.com/pytorch/pytorch/pull/14500

    简而言之:使用 at::cuda::getCurrentCUDAStream()