facebookresearch/HolisticTraceAnalysis
Support the Equivalent of profiler.key_averages(group_by_stack_n=n)
Ouverte
#294 ouverte le 3 sept. 2025
help wantedneeds triagequestion
Métriques du dépôt
- Stars
- (535 étoiles)
- Métriques de merge PR
- (Métriques PR en attente)
Description
What is your question?
The pytorch profiler can aggregate runtimes by the operation and its traceback via key_averages(group_by_stack_n=n): https://docs.pytorch.org/tutorials/beginner/profiler.html#print-profiler-result
Identifying the top operators / kernels and their source user Python code is helpful to find the main bottlenecks and perform targeted optimizations. Does HTA support something equivalent, or could you shed some light on implementing one? Thanks!
I also found get_aten_op_kernels_and_delay(), but seems like it traces down from aten ops only instead of source user training code.
Code
No response
What have you tried?
No response
Environment
No response