StreamContext

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

Менеджер контекста, выбирающий заданный поток.

Все ядра XPU, поставленные в очередь в его контексте, будут добавлены в очередь выбранного потока.

Параметры:

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

Примечание

Потоки привязаны к устройству.