torch.mps.profiler.start

torch.mps.profiler.start(mode='interval', wait_until_completed=False) [исходный код]

Запускает трассировку OS Signpost из бэкенда MPS.

Созданные OS Signpost можно записывать и просматривать в инструменте Logging приложения Xcode Instruments.

Параметры:

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