Function torch::mps::synchronize

• Defined in File mps.h

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.