快捷方式

torch.xpu.get_stream_from_external

torch.xpu.get_stream_from_external(data_ptr, device=None)[source][source]

從外部 SYCL 佇列返回一個 Stream 物件。

此函式用於包裝在其他庫中建立的 SYCL 佇列,以便促進資料交換和多庫互動。

注意

此函式不管理佇列的生命週期,使用者有責任在使用返回的 Stream 物件期間保持引用的佇列處於活躍狀態。不同的 SYCL 佇列指標將導致不同的 Stream 物件,即使它們解引用的 SYCL 佇列是等效的。

引數
  • data_ptr (int) – 外部傳入的 sycl::queue* 值的整數表示。

  • device (torch.deviceint,可選) – 佇列最初建立所在的裝置。使用者有責任確保裝置指定正確。

返回型別

Stream

文件

訪問 PyTorch 的全面開發者文件

檢視文件

教程

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

檢視教程

資源

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

檢視資源