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) – Ожидает завершения выполнения потока MPS для каждой кодируемой операции GPU. Это помогает генерировать единичные диспетчеризации на временной шкале трассировки. Обратите внимание, что включение этого параметра отрицательно скажется на производительности.