Rate this Page

torch.mtia.synchronize#

torch.mtia.synchronize(device=None)[source]#

Waits for all jobs in all streams on a MTIA device to complete.