torch.cuda.get_stream_from_external

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

Возвращает Stream для внешнего потока CUDA.

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

Примечание

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

Параметры:

• data_ptr (int) – целочисленное представление значения cudaStream_t, выделенного извне.
• device (torch.device or int, optional) – устройство, на котором изначально был выделен поток. Если устройство указано неверно, последующие запуски с использованием этого потока могут завершиться ошибкой.

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

Stream