Try to modify the following content in the files which prompt the errors above.
#include <ATen/cuda/CUDAContext.h>
// #include <THC/THC.h>
#include <cuda.h>
#include <cuda_runtime_api.h>
#include “ball_query_gpu.h”
// extern THCState *state;
Try to modify the following content in the files which prompt the errors above.
#include <ATen/cuda/CUDAContext.h>
// #include <THC/THC.h>
#include <cuda.h>
#include <cuda_runtime_api.h>
#include “ball_query_gpu.h”
// extern THCState *state;