Compatibility of torch>=1.5.0

[ErictheSam]

My torch version is 1.9.0, and caught errors within the following two functions/macros: AT_CHECK and THCState_getCurrentStream(state) I changed them to TORCH_CHECK and at::cuda::getCurrentCUDAStream().stream() and it worked well, so I wonder if those changes can be published in order to support later versions.