torch.mps.synchronize

torch.mps.synchronize()[源代码]

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