Событие

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

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

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

Параметры:

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

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

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

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

float

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

Возвращает True, если все операции, на данный момент отслеживаемые событием, завершены.

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

bool

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

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

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

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

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

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