torch.mtia.synchronize

torch.mtia.synchronize(device=None)

等待 MTIA 裝置上所有流中的所有作業完成。