torch.mtia.set_stream

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

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

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

Параметры:

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