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.