Module View doesn't show backward pass operations

[danielsnider]

The backward pass doesn't appear in the Module View. Is there a plan to add this?

The two red arrows below mark where the backward pass should appear:

Here is the backward pass as seen in the Trace View:

This profile was generated using your sample script resnet50_profiler_api.py and with python3.10, pytorch 1.12.0.dev20220408+cu113, and Ubuntu 20.04.

Many thanks for your great work on Kineto!

[briancoutinho]

Is this a feature request for tensorboard plugin? looks like the trace from libkineto has the information right?

[guotuofeng]

@danielsnider, there is a bug in PyTorch profiler related to forward/backward correlation which would cause the profiler crash. We deliberately disable the forward/backward correlation in nightly build. Once the bug is fixed, the forward/backward correlation would be reeanbled again. At that time, the backward module would be shown in tb_plugin.

[SolenoidWGT]

Hello @guotuofeng , Is there any progress on this bug?