Событие

class torch.xpu.Event(enable_timing=False) [исходный код]

Обёртка для события XPU.

События XPU — это маркеры синхронизации, которые можно использовать для отслеживания хода выполнения на устройстве и синхронизации потоков XPU.

Базовые события XPU инициализируются отложенно при первой записи события. После создания записывать событие могут только потоки на том же устройстве. Однако потоки на любом устройстве могут ожидать это событие.

Параметры:

enable_timing (bool, необязательно) – указывает, должно ли событие измерять время (по умолчанию: False)

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

Возвращает прошедшее время.

Время в миллисекундах между записью события и записью end_event.

Параметры:

end_event (Event) – конечное событие.

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

Проверяет, завершена ли вся работа, на данный момент отслеживаемая событием.

Возвращает:

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

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

bool

record(stream=None) [исходный код]

Записывает событие в указанный поток.

Параметры:

stream (Stream, torch.Stream, необязательно) – Используется torch.xpu.current_stream(), если поток не указан. Устройство потока должно совпадать с устройством события.

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

Ожидает завершения события.

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

wait(stream=None) [исходный код]

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

Параметры:

stream (Stream, torch.Stream, необязательно) – Используется torch.xpu.current_stream(), если поток не указан.