Поток

class torch.xpu.Stream(device=None, priority=0, **kwargs) [исходный код]

Обёртка над потоком XPU.

Поток XPU — это линейная последовательность выполнения, относящаяся к определённому устройству и независимая от других потоков. Он поддерживает использование оператора with в качестве менеджера контекста, чтобы гарантировать, что операторы внутри блока with выполняются в соответствующем потоке.

Параметры:

• device (torch.device or int, optional) — устройство, на котором будет создан поток. Если device равно None (по умолчанию) или отрицательному целому числу, будет использоваться текущее устройство.
• priority (int, optional) — приоритет потока, который может быть положительным, равным 0 или отрицательным. Чем меньше число, тем выше приоритет. По умолчанию приоритет равен 0. Если значение выходит за допустимый диапазон приоритетов, оно автоматически будет сопоставлено с ближайшим допустимым приоритетом (наименьшим для больших положительных чисел или наибольшим для больших отрицательных чисел).

query() [исходный код]

Проверить, завершена ли вся отправленная работа.

Возвращает:

Логическое значение, указывающее, завершены ли все ядра в этом потоке.

Тип возвращаемого значения:

bool

record_event(event=None) [исходный код]

Записать событие.

Параметры:

event (Event, torch.Event, optional) — событие для записи. Если оно не задано, будет создано новое.

Возвращает:

Записанное событие.

synchronize() [исходный код]

Дождаться завершения всех ядер в этом потоке.

wait_event(event) [исходный код]

Заставить всю будущую работу, отправленную в поток, ожидать наступления события.

Параметры:

event (Event, torch.Event) — событие, которого нужно дождаться.

wait_stream(stream) [исходный код]

Синхронизироваться с другим потоком.

Вся будущая работа, отправленная в этот поток, будет ожидать завершения всех ядер, отправленных в указанный поток на момент вызова.

Параметры:

stream (Stream, torch.Stream) — поток для синхронизации.