torch.xpu.synchronize#
- torch.xpu.synchronize(device=None)[source]#
等待XPU设备上的所有流中的所有内核完成。
- 参数
device (torch.device 或 int, optional) – 要同步的设备。如果
device为None(默认)，则使用当前设备，由current_device()指定。
等待XPU设备上的所有流中的所有内核完成。
device (torch.device 或 int, optional) – 要同步的设备。如果 device 为 None (默认)，则使用当前设备，由 current_device() 指定。