Function torch::cuda::synchronize

• Defined in File cuda.h

Function Documentation

void torch::cuda::synchronize(int64_t device_index = -1)

Waits for all kernels in all streams on a CUDA device to complete.