Shortcuts

Function torch::mps::synchronize

Function Documentation

void torch::mps::synchronize()

Waits for all streams on the MPS device to complete.

This blocks the calling CPU thread by using the ‘waitUntilCompleted()’ method to wait for Metal command buffers finish executing all the encoded GPU operations before returning.

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources