torch.mps.profiler.metal_capture

torch.mps.profiler.metal_capture(fname)[source][source]

Conext manager that enables capturing of Metal calls into gputrace