facebookresearch/HolisticTraceAnalysis
Is it possible to categorize the kernels befween forward and backward passes
オープン
#206 opened on 2024/11/29
help wantedneeds triagequestion
Repository metrics
- Stars
- (535 個のスター)
- PR merge metrics
- (PR metrics pending)
説明
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