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