torch.cuda.set_stream

torch.cuda.set_stream(stream) [исходный код]

Установить текущий поток. Это API-обёртка для установки потока.

Использование этой функции не рекомендуется; вместо неё следует использовать менеджер контекста stream.

Параметры:

stream (Stream) – выбранный поток. Эта функция ничего не делает, если этот аргумент равен None.