已合并
fix benchmark_gpu bug #44978
xinyuzhuo创建于 7 小时前
fix benchmark_gpu bug #44978
已合并
xinyuzhuo创建于 7 小时前
xinyuzhuo
7 小时前

【合入来源】

如有社区issue,请关联issue链接
请勿携带内部流程信息(需求链接、问题单、内部issue等)
https://gitcode.com/Ascend/pytorch/issues/4239

【修改方案】

请描述修改内容的具体实现,涉及哪些组件之间进行交互,可以用1、2、3、...进行罗列
如果是需求或者重构类的PR,需要补充详细设计文档(说明上下游组件关系、时序图、类图、DFX能力等内容)

【资料变更】

不涉及

【接口变更】

不涉及

【功能验证】

【CheckList】

PR提交人对以下CheckList自检项进行全量自检,自检通过或不涉及,均修改 [ ] 为 [x]

likedislike
Pull Request已成功合入, 合并人@ascend-robot
(感谢 xinyuzhuo 的贡献)
Xxinyuzhuo
7 小时前 创建了 pull request,commit ad2a8d7a
atomgit-bot
atomgit-bot
7 小时前 评论:

变更摘要

本 PR 修复了 torch_npu 中 benchmark_gpu 的 bug,改动集中在 torch_npu/_inductor/runtime/triton_heuristics.pyNPUCachingAutotuner 类内:在调用 benchmarker.benchmark_gpu(kernel_call, rep=1) 时新增 is_vetted_benchmarking=True 参数,将该路径下的基准测试标记为已验证(vetted)模式,以修正原有调用方式存在的问题。

主要改动

  • 新增 is_vetted_benchmarking=True 参数:将 benchmark_gpu(kernel_call, rep=1) 改为 benchmark_gpu(kernel_call, rep=1, is_vetted_benchmarking=True),为该基准测试调用补充已验证标记。
  • 定位修复范围:该改动位于 NPUCachingAutotuner 类中,仅影响走 benchmarker.benchmark_gpu 的基准测试分支;其前序的 do_bench_using_profiling_npu(kernel_call, rep=1) 分支调用保持不变。
likedislike
ascend-robot
ascend-robot成员
7 小时前 评论:
atomgit-bot
atomgit-bot
7 小时前 评论:

代码审查

✅ 未发现问题

likedislike
ascend-robotascend-robot成员
7 小时前 添加了label:ascend-cla/yes
此处折叠了57条消息 查看更多
dezheng889成员
4 小时前 评论:

/approve

likedislike
ascend-robotascend-robot成员
4 小时前 添加了label:approved
ascend-robotascend-robot成员
4 小时前 关闭了关联的issue
ascend-robotascend-robot成员
4 小时前 合入了pull request
ascend-robot
ascend-robot成员
4 小时前 评论:
流水线 pytorch_gitcode_PR_multiVersion#14308 [ commitID:a448b1f0 ] 已完成
likedislike