pretrained model

[SeyedHamidreza]

Hi, Thank you for sharing the code.

For the newer version of CUDA, we need to replace THCudaCheck with AT_CUDA_CHECK

//THCudaCheck(cudaGetLastError()); AT_CUDA_CHECK(cudaGetLastError());

and we need to comment //THArgCheck(at::cuda::getApplyGrid(totalElements, grid, curDevice), 1, "Too many elements to calculate");

in sampling_kernel.cu and ball_query_kernel.cu we need to do the following

//#include <THC/THC.h> #include <ATen/cuda/CUDAContext.h> #include <ATen/cuda/CUDAEvent.h>

for more info, please check https://github.com/pytorch/pytorch/issues/72807

best regards, Hamidreza