我正在努力跟上 this tutorial
明确地
THCState *state 论点——这个论点来自哪里?我怎样才能得到一份工作 state 例如,在cuda中实现 tensor.cat :
THCState *state
state
tensor.cat
void THCTensor_(cat)(THCState *state, THCTensor *result, THCTensor *ta, THCTensor *tb, int dimension)
tensor.cat() python中没有提供任何
tensor.cat()
状态 然后转换为 cudaStream_t stream = THCState_getCurrentStream(state); 出于某种原因, THCState_getCurrentStream 不再有定义了吗?我怎样才能拿到票 stream 从我的 状态 ?
状态
cudaStream_t stream = THCState_getCurrentStream(state);
THCState_getCurrentStream
stream
我还试着继续问 pytorch 论坛-到目前为止没有结果。
它已弃用(没有文档!) 请看这里: https://github.com/pytorch/pytorch/pull/14500
简而言之:使用 at::cuda::getCurrentCUDAStream()
at::cuda::getCurrentCUDAStream()