已合并
foreach_add_listV2算子tiling bug修复 #7398
zhaozhongyao创建于 7月10日
foreach_add_listV2算子tiling bug修复 #7398
已合并
zhaozhongyao创建于 7月10日
zhaozhongyao
zhaozhongyao成员
7月10日

描述

foreach_add_listV2算子tiling bug导致kernel侧重复计算,出现精度问题

关联的Issue

https://gitcode.com/cann/ops-nn/issues/4039

测试

已验证精度pass

文档更新

类型标签

AI/Agent生成声明

likedislike
Pull Request已成功合入, 合并人@CANN-robot
(感谢 zhaozhongyao 的贡献)
zhaozhongyaozhaozhongyao成员
7月10日 创建了 pull request,commit 06bb948d
atomgit-bot
atomgit-bot
7月10日 评论:

变更摘要

本次变更修复了 ForeachAddList 算子在 tiling 阶段的一个 bug:在 perCoreElementsALIGN_SIZE 向上对齐后,原先的 needCoreNum 不再与对齐后的每核元素数匹配,导致可能出现多余的核参与计算。修复方式是在 perCoreElements 对齐后重新计算 needCoreNum,使核数与对齐后的分块大小保持一致。

主要改动

  • ForeachAddListTilingFunc 中重新计算 needCoreNum:在对 perCoreElements 进行 ALIGN_SIZE 对齐后,新增一行 needCoreNum = Ops::Base::CeilDiv(totalElements, perCoreElements),确保所需的核数基于对齐后的每核元素数重新计算,避免因对齐放大导致核数冗余。
likedislike
atomgit-bot
atomgit-bot
7月10日 评论:

代码审查

✅ 未发现问题

likedislike
CANN-robotCANN-robot成员
7月10日 添加了label:cann-cla/yes
CANN-robot
CANN-robot成员
7月10日 评论:

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
** gcw_kUomxQ2l, 杨阳 (2/2) gcw_kUomxQ2l (1/1)

💡 Tip:

  • Committer can comment /approve or /lgtm
  • Commenting /approve implies both code review (lgtm) and intent to merge (approve)

CLA Signature Pass

zhaozhongyao, thanks for your pull request. All authors of the commits have signed the CLA. 👍

likedislike
此处折叠了71条消息 查看更多
CANN-robotCANN-robot成员
7月11日 添加了label:approved
yangyang016成员
7月11日 评论:

/lgtm

likedislike
CANN-robotCANN-robot成员
7月11日 添加了label:lgtm
CANN-robotCANN-robot成员
7月11日 关闭了关联的issue
CANN-robotCANN-robot成员
7月11日 合入了pull request