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