Synchronize CUDA calls in Libtorch

[ptrblck]

This should work:

CUDAStream stream = getCurrentCUDAStream();
AT_CUDA_CHECK(cudaStreamSynchronize(stream));