torch.mps.profiler.profile

torch.mps.profiler.profile(mode='interval', wait_until_completed=False) [source]

Менеджер контекста для включения создания трассировки OS Signpost из бэкенда MPS.

Параметры:

• mode (str) – Режим трассировки OS Signpost может быть «interval», «event» или «interval,event». Режим interval отслеживает длительность выполнения операций, а режим event отмечает завершение выполнения. Дополнительную информацию см. в документе Запись данных о производительности.
• wait_until_completed (bool) – Ожидает завершения выполнения каждой закодированной операции GPU в потоке MPS. Это помогает отображать отдельные отправки на временной шкале трассировки. Обратите внимание, что включение этой опции негативно повлияет на производительность.

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

Iterator[None]