Profile backward pass of DNN

Hi, I wanted to profile the per layer execution times for backward pass of a DNN. Existing profilers like pytorch profiler give operator-level stats. But, I wanted a layerwise time. So, I wanted to know if Nvidia has a profiler to do this ?
Does DLPROF : DLProf User Guide - NVIDIA Docs help with this ?

Device : Orin Jetpack v5.0.1
Using : Pytorch v2.0.0

@dusty_nv Would you be able to help please?

Hi @vinayakah ,
Apologies for the delay.
Let me check on this, will get back to you shortly.


1 Like