sshaoshuai / Pointnet2.PyTorch

A faster implementation of PointNet++ based on PyTorch.
MIT License
476 stars 107 forks source link

THC\THC.h can't find #34

Open zhou-SHU164 opened 2 years ago

zhou-SHU164 commented 2 years ago

1648721360(1) Did anyone encounter this problem and know how to fix it ?

0vvO commented 2 years ago

pytorch(1.11) has removed THC.h

zhou-SHU164 commented 2 years ago

pytorch(1.11) has removed THC.h

I have found it. Thank you so much!

flandrewries commented 2 years ago

//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

fusturnu commented 2 years ago

//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.

tdcsu commented 11 months ago

//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

HenryWJL commented 3 months ago

This solution also works for me, thanks.

tdcsu commented 3 months ago

您好,我已收到您的邮件,会尽快处理~

inkel-x commented 1 month ago

//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.

tdcsu commented 1 month ago

您好,我已收到您的邮件,会尽快处理~