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.device 或 int,可選) – 佇列最初建立所在的裝置。使用者有責任確保裝置指定正確。
- 返回型別