已合并
fix benchmark_gpu bug #44978
xinyuzhuo创建于 7 小时前
fix benchmark_gpu bug #44978
已合并
Pull Request已成功合入, 合并人@ascend-robot
(感谢 xinyuzhuo 的贡献)atomgit-bot
7 小时前 评论:
7 小时前 评论:
变更摘要
本 PR 修复了 torch_npu 中 benchmark_gpu 的 bug,改动集中在 torch_npu/_inductor/runtime/triton_heuristics.py 的 NPUCachingAutotuner 类内:在调用 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)分支调用保持不变。


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


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


【合入来源】
【修改方案】
【资料变更】
不涉及
【接口变更】
不涉及
【功能验证】
【CheckList】