PyTorch Profiler Per Layer

1

Hi there,

Just wonder the current PyTorch Profiler (1.8.1+) supports the per layer profiling other than the lower-level operator profiling?

I check the documentation (torch.profiler — PyTorch 1.8.1 documentation) and looks like in no place it mentions per-layer operations.

Thanks!

2

I have the same question, did you find any answer yet?