Open Chuxwa opened 3 years ago
I replaced all mentions of AT_CHECK and THCState_getCurrentStream(state) with TORCH_CHECK and at::cuda::getCurrentCUDAStream(), respectively. I was then able to install setup.py without errors.
I replaced all mentions of AT_CHECK and THCState_getCurrentStream(state) with TORCH_CHECK and at::cuda::getCurrentCUDAStream(), respectively. I was then able to install setup.py without errors.