Rate this Page

torch.autograd.profiler.load_nvprof#

torch.autograd.profiler.load_nvprof(path)[source]#

Open an nvprof trace file and parses autograd annotations.

Parameters

path (str) – path to nvprof trace