快捷方式

torch.xpu.synchronize

torch.xpu.synchronize(device=None)[source][source]

等待 XPU 裝置上所有流中的所有核完成執行。

引數

device (torch.device or int, optional) – 要同步的裝置。如果 deviceNone (預設),則使用 current_device() 指定的當前裝置。

文件

查閱 PyTorch 的全面開發者文件

檢視文件

教程

獲取面向初學者和高階開發者的深度教程

檢視教程

資源

查詢開發資源並獲得問題解答

檢視資源