Shortcuts

torch.mps.profiler.profile

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

Context Manager to enabling generating OS Signpost tracing from MPS backend.

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.

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources