Событие

class torch.mps.event.Event(enable_timing=False) [source]

Обёртка над событием MPS.

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

Параметры

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

elapsed_time(end_event) [source]

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

query() [source]

Возвращает True, если вся работа, в настоящее время захваченная событием, завершена.

record() [source]

Записывает событие в поток по умолчанию.

synchronize() [source]

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

wait() [source]

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