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，可选) – 队列最初创建所在的设备。用户有责任确保设备指定正确。

返回类型

Stream