Rate this Page
★ ★ ★ ★ ★

torch.mps.profiler.metal_capture#

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

Conext manager that enables capturing of Metal calls into gputrace

On this page
Show Source
PyTorch Community
Language Bindings
PyTorch Libraries