torch.xpu.set_stream

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

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

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

Параметры:

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