torch.mps.synchronize

torch.mps.synchronize()

等待 MPS 设备上的所有流中的所有内核完成。