torch.mps.synchronize

torch.mps.synchronize()

等待 MPS 裝置上的所有流中的所有核心完成。