torch.mps.profiler.start

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

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

Сгенерированные OS Signpost можно записать и просмотреть в инструменте XCode Instruments Logging.

Параметры

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