已合并
修改reducesum GetInputParam函数调用 #4157
sunzhongwen1创建于 25 天前
修改reducesum GetInputParam函数调用 #4157
已合并
Pull Request已成功合入, 合并人@CANN-robot
(感谢 sunzhongwen1 的贡献)25 天前 创建了 pull request,commit 749fb570
atomgit-bot
25 天前 评论:
25 天前 评论:
变更摘要
此变更修复了 ReduceSum 算子在 arch35 架构下的 batch 一致性(batch invariance)问题。核心改动是将 ReduceTilingKey 的构造和 batchInvariant 标志的设置提前到 ReduceOpTmpl::GetInputParam 调用之前,并将 key 作为新参数传入该函数,使得确定性级别信息在获取输入参数阶段即生效,而非仅在后续 DoTiling 阶段才使用。
主要改动
GetInputParam调用增加key参数:ReduceOpTmpl::GetInputParam(context, opInput, 0, 1, 0)变更为ReduceOpTmpl::GetInputParam(context, opInput, 0, 1, 0, key),将ReduceTilingKey传入以便在输入参数获取阶段即可感知 batch 一致性设置。ReduceTilingKey初始化时机前移: 将ReduceTilingKey key的声明及batchInvariant标志(基于context->GetDeterministicLevel() == 2判断)的设置从GetInputParam调用之后移动到调用之前,确保key在传入时已正确配置。


atomgit-bot
25 天前 评论:
25 天前 评论:
25 天前 添加了label:cann-cla/yes
CANN-robot
25 天前 评论:
25 天前 评论:
Thanks for your pull-request.
The full list of commands accepted by me can be found at here。
You can get sig-info at here
PR Approval Progress
✅ Congratulations! All modules have met the lgtm and approve requirements.
Module Approval Details
| module | lgtm status | approve status |
|---|---|---|
| repo-cann/ops-math | ✅ jia0liang, wangrui_ (2/2) | ✅ jia0liang (1/1) |
💡 Tip:
- Committer can comment
/approveor/lgtm- Commenting
/approveimplies both code review (lgtm) and intent to merge (approve)
CLA Signature Pass
sunzhongwen1, thanks for your pull request. All authors of the commits have signed the CLA. 👍


此处折叠了70条消息 查看更多
23 天前 添加了label:ci-pipeline-passed
23 天前 添加了label:lgtm
23 天前 关闭了关联的issue
23 天前 合入了pull request
描述
base仓新增用于支持batch一致性算子的GetInputParam函数,算子侧做对应修改。
关联的Issue
测试
文档更新
类型标签