facebookresearch/HolisticTraceAnalysis
Is it possible to categorize the kernels befween forward and backward passes
Offen
#206 geöffnet am 29.11.2024
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