torch_sys::cuda

Function 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.