StreamContext

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

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

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

Параметры:

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

Примечание

Потоки существуют отдельно для каждого устройства.