torch.xpu.synchronize¶
- torch.xpu.synchronize(device=None)[源代码]¶
等待XPU设备上所有流中的所有内核完成。
- Parameters
设备 (torch.device 或 整数, 可选) – 要同步的设备。 如果
device为None（默认），则使用由current_device()给出的当前设备。
等待XPU设备上所有流中的所有内核完成。
设备 (torch.device 或 整数, 可选) – 要同步的设备。
如果 device 为 None（默认），则使用由 current_device() 给出的当前设备。