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.

On this page
Show Source
PyTorch Libraries