facebookresearch/HolisticTraceAnalysis

Is it possible to categorize the kernels befween forward and backward passes

Offen

#206 geöffnet am 29.11.2024

 (1 Kommentar) (0 Reaktionen) (0 zugewiesene Personen)Python (94 Forks)auto 404
help wantedneeds triagequestion

Repository-Metriken

Stars
 (535 Sterne)
PR-Merge-Metriken
 (PR-Metriken ausstehend)

Beschreibung

What is your question?

Thanks for the great library!

analyzer.get_gpu_kernel_breakdown(visualize=True) returns a dataframe with kernels statistics. Is it possible to split those between forward/backward categories?

Code

No response

What have you tried?

No response

Environment

No response

Contributor Guide