torch.xpu.get_stream_from_external

torch.xpu.get_stream_from_external(data_ptr, device=None) [исходный код]

Возвращает Stream из внешней очереди SYCL.

Эта функция используется для обёртывания очередей SYCL, созданных в других библиотеках, чтобы упростить обмен данными и взаимодействие между библиотеками.

Примечание

Эта функция не управляет жизненным циклом очереди; пользователь несёт ответственность за то, чтобы указанная очередь оставалась действующей, пока используется возвращённый поток. Разные указатели на очереди SYCL будут соответствовать разным объектам Stream, даже если очереди SYCL, на которые они указывают, эквивалентны.

Параметры:

• data_ptr (int) – целочисленное представление значения sycl::queue*, переданного извне.
• device (torch.device или int, необязательно) – устройство, на котором изначально была создана очередь. Пользователь несёт ответственность за правильное указание устройства.

Тип возвращаемого значения:

Stream