torch.xpu.get_stream_from_external#
- torch.xpu.get_stream_from_external(data_ptr, device=None)[原始碼]#
從外部 SYCL 佇列返回一個
Stream。此函式用於封裝在其他庫中建立的 SYCL 佇列，以促進資料交換和多庫互動。
注意
此函式不管理佇列的生命週期，使用者有責任在返回的 stream 使用期間保持引用的佇列處於活動狀態。不同的 SYCL 佇列指標將導致不同的
Stream物件，即使它們引用的 SYCL 佇列是等效的。- 引數
data_ptr (int) – 外部傳入的 sycl::queue* 值的整數表示。
device (torch.device 或 int, 可選) – 佇列最初建立的裝置。使用者有責任確保裝置指定正確。
- 返回型別
