Rate this Page
★ ★ ★ ★ ★

torch.mps.profiler.metal_capture#

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

Context manager that enables capturing of Metal calls into gputrace

Return type:

Iterator[None]

On this page
Show Source
PyTorch Libraries