error: ‘getCurrentCUDAStream’ is not a member of ‘at::cuda’ cudaStream_t stream = at::cuda::getCurre

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;