torch.mtia.synchronize

torch.mtia.synchronize(device=None)[源代码]

等待MTIA设备上所有流中所有作业的完成。