facebookresearch/HolisticTraceAnalysis

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

开放

#206 创建于 2024年11月29日

在 GitHub 查看贡献者指南
 (1 条评论) (0 个反应) (0 位负责人)Python (94 个派生)auto 404
help wantedneeds triagequestion

仓库指标

星标
 (535 个星标)
PR 合并指标
 (PR 指标待抓取)

描述

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

贡献者指南

研究方向
调查现有的GPU内核分解功能，并探索使用PyTorch的自动梯度或分析器钩子按传递（前向/反向）标记内核的方法。
技术栈
pythonpytorch
领域
machine learningperformance
议题类型
调研
难度面向新贡献者的预计实现难度，1 表示很小改动，5 表示专家级工作。
3
预计时间有经验贡献者完成调查、实现、测试并准备 pull request 的粗略时间范围。
半天
活动状态议题当前的可参与程度：新鲜、活跃、陈旧、阻塞或等待维护者输入。
活跃
清晰度议题是否清楚说明期望改动、验收标准和下一步。
清晰
前置要求
PythonPyTorch
新手友好度1-100 的估计分数，表示该议题对首次贡献者的友好程度。
60