Hi all,
I am trying to implement a mixed C++ CUDA Torch extension. My understanding is that I can only use ATen.h for this.  (e.g. Torch.h cannot be used). The issue is that I wan to synchronize CUDA kernels from CPU, so I am trying to do the equivalent to cudaDeviceSynchronize() in CUDA , I  cannot find any similar function within at::cuda class.   Could anyone help me out please ?  Am I missing something ?