Rate this Page
★ ★ ★ ★ ★

torch.mps.synchronize#

torch.mps.synchronize()[source]#

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

On this page
Show Source
PyTorch Libraries