已合并
修改reducesum GetInputParam函数调用 #4157
修改reducesum GetInputParam函数调用 #4157
已合并
sunzhongwen1创建于 25 天前
sunzhongwen1
sunzhongwen1成员
25 天前

描述

base仓新增用于支持batch一致性算子的GetInputParam函数,算子侧做对应修改。

关联的Issue

测试

文档更新

类型标签

  • Bug修复
  • 新特性
  • 性能优化
  • 文档更新
  • 其他,请描述:
likedislike
Pull Request已成功合入, 合并人@CANN-robot
(感谢 sunzhongwen1 的贡献)
sunzhongwen1sunzhongwen1成员
25 天前 创建了 pull request,commit 749fb570
atomgit-bot
atomgit-bot
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 在传入时已正确配置。
likedislike
atomgit-bot
atomgit-bot
25 天前 评论:

代码审查

✅ 未发现问题

likedislike
CANN-robotCANN-robot成员
25 天前 添加了label:cann-cla/yes
CANN-robot
CANN-robot成员
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 /approve or /lgtm
  • Commenting /approve implies 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. 👍

likedislike
此处折叠了70条消息 查看更多
CANN-robotCANN-robot成员
23 天前 添加了label:ci-pipeline-passed
RuiWang_成员
23 天前 评论:

/lgtm

likedislike
CANN-robotCANN-robot成员
23 天前 添加了label:lgtm
CANN-robotCANN-robot成员
23 天前 关闭了关联的issue
CANN-robotCANN-robot成员
23 天前 合入了pull request