Rate this Page

torch.mps.synchronize#

torch.mps.synchronize()[source]#

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