Open zhou-SHU164 opened 2 years ago
pytorch(1.11) has removed THC.h
pytorch(1.11) has removed THC.h
I have found it. Thank you so much!
//Comment Out /src/ball_query.cpp group_points.cpp interpolate.cpp pointnet2_api.cpp sampling.cpp //#include <THE/THC.h> //extern THCState *state; //cudaStream_t stream = THCState_getCurrentStream(state);
//Replace with
cudaStream_t stream = at::cuda::getCurrentCUDAStream();
then replace the AT_CHECK with TORCH_CHECK in the ball_query.cpp
//Comment Out /src/ball_query.cpp group_points.cpp interpolate.cpp pointnet2_api.cpp sampling.cpp //#include <THE/THC.h> //extern THCState *state; //cudaStream_t stream = THCState_getCurrentStream(state);
//Replace with #include <ATen/cuda/CUDAContext.h> #include <ATen/cuda/CUDAEvent.h> cudaStream_t stream = at::cuda::getCurrentCUDAStream();
then replace the AT_CHECK with TORCH_CHECK in the ball_query.cpp
This solution proposal worked for me. Thanks.
//Comment Out /src/ball_query.cpp group_points.cpp interpolate.cpp pointnet2_api.cpp sampling.cpp //#include <THE/THC.h> //extern THCState *state; //cudaStream_t stream = THCState_getCurrentStream(state);
//Replace with #include <ATen/cuda/CUDAContext.h> #include <ATen/cuda/CUDAEvent.h> cudaStream_t stream = at::cuda::getCurrentCUDAStream();
then replace the AT_CHECK with TORCH_CHECK in the ball_query.cpp
This solution works for me, thanks
This solution also works for me, thanks.
您好,我已收到您的邮件,会尽快处理~
//Comment Out /src/ball_query.cpp group_points.cpp interpolate.cpp pointnet2_api.cpp sampling.cpp //#include <THE/THC.h> //extern THCState *state; //cudaStream_t stream = THCState_getCurrentStream(state);
//Replace with #include <ATen/cuda/CUDAContext.h> #include <ATen/cuda/CUDAEvent.h> cudaStream_t stream = at::cuda::getCurrentCUDAStream();
then replace the AT_CHECK with TORCH_CHECK in the ball_query.cpp
it works! Thanks.
您好,我已收到您的邮件,会尽快处理~
Did anyone encounter this problem and know how to fix it ?