torch.mps.profiler.metal_capture

torch.mps.profiler.metal_capture(fname)

Conext manager that enables capturing of Metal calls into gputrace