Function torch_sys::cuda::atc_synchronize
source · pub unsafe extern "C" fn atc_synchronize(device_index: i64)Expand description
Waits for all kernels in all streams on a CUDA device to complete.
pub unsafe extern "C" fn atc_synchronize(device_index: i64)Waits for all kernels in all streams on a CUDA device to complete.