On this page
torch.mps.profiler.start
torch.mps.profiler.start(mode='interval', wait_until_completed=False)
[source]-
Start OS Signpost tracing from MPS backend.
The generated OS Signposts could be recorded and viewed in XCode Instruments Logging tool.
- Parameters
-
- mode (str) – OS Signpost tracing mode could be “interval”, “event”, or both “interval,event”. The interval mode traces the duration of execution of the operations, whereas event mode marks the completion of executions. See document Recording Performance Data for more info.
- wait_until_completed (bool) – Waits until the MPS Stream complete executing each encoded GPU operation. This helps generating single dispatches on the trace’s timeline. Note that enabling this option would affect the performance negatively.
© 2024, PyTorch Contributors
PyTorch has a BSD-style license, as found in the LICENSE file.
https://pytorch.org/docs/2.1/generated/torch.mps.profiler.start.html