THCState_getCurrentStream in torch1.5

torch 1.4에서 썼었던 THCState_getCurrentStream은 torch 1.5로 변경되면서 at::cuda::getCurrentCUDAStream().stream()로 바뀌었다 orginal error: /ops/deformable_convolution/src/cuda/dcn_v2_cuda.cu(107): error: identifier "THCState_getCurrentStream" is undefined /ops/deformable_convolution/src/cuda/dcn_v2_cuda.cu(279): error: identifier "THCState_getCurrentStream" is undefined 2 errors detected in the compilation of "/tmp/tmpxft_0000125b_00000000-6_dcn_v2_cuda.cpp1.ii". I change createBatchGemmBuffer<<<grid, block, 0, THCState_getCurrentStream(state)>>&gt..........

원문링크 : THCState_getCurrentStream in torch1.5