Use at::cuda::CUDAStream compatible interfaces
#10
+106
−88
at::cuda::CUDAStream compatible interfaces
#10