torch.xpu.synchronize¶
- torch.xpu.synchronize(device=None)[原始碼][原始碼]¶
等待 XPU 裝置上所有串流中的所有核心完成。
- 參數
device (torch.device 或 int, optional) – 要同步的裝置。如果
device
為None
(預設),則使用current_device()
給定的目前裝置。
等待 XPU 裝置上所有串流中的所有核心完成。
device (torch.device 或 int, optional) – 要同步的裝置。如果 device
為 None
(預設),則使用 current_device()
給定的目前裝置。