已合并
feat(arch35): add Ascend 950 support for Adam optimizer operators #8078
daiwei18创建于 7月30日
feat(arch35): add Ascend 950 support for Adam optimizer operators #8078
已合并
daiwei18创建于 7月30日
daiwei18
daiwei18
7月30日

描述

  • 为 ApplyAdamWQuant 增加 arch35/Ascend 950 tiling、kernel、binary 配置和测试覆盖。
  • 新增 ApplyAdamWithAmsgrad 算子目录,包含 proto、host、kernel、Ascend950 配置、示例和 UT。
  • 修复 ApplyAdamWQuant 公共 tiling 头文件命名和 UT tiling 类型对齐问题。
  • 补充 README、aclnn 文档和 golden 资产,完善算子提交材料。

关联的Issue

关联Issue #4461
https://gitcode.com/cann/ops-nn/issues/4461

测试

  • 新增/更新 ApplyAdamWQuant、ApplyAdamWithAmsgrad arch35 op_host/op_kernel UT。
  • ApplyAdamWQuant golden 入口本地可执行;dtype 与校验反例覆盖已补充。
  • 本地执行 clang-format 18.1.8 --dry-run --Werror,改动 C/C++ 文件格式检查通过。
  • 精度、泛化和性能结果随转测补充。

文档更新

  • 更新 ApplyAdamWQuant 文档,并新增 ApplyAdamWithAmsgrad README/示例说明。

类型标签

AI/Agent生成声明

likedislike
Pull Request已成功合入, 合并人@CANN-robot
(感谢 daiwei18 的贡献)
daiwei18daiwei18
7月30日 创建了 pull request,commit 4db40487
atomgit-bot
atomgit-bot
7月30日 评论:

🤖 正在生成合并请求摘要,请稍候…

likedislike
不准确?
atomgit-bot
atomgit-bot
7月30日 评论:

🤖 AI 代码检视正在进行中,请稍候…

likedislike
不准确?
CANN-robotCANN-robot成员
7月30日 添加了label:cann-cla/yes
CANN-robotCANN-robot成员
7月30日 添加了label:stat/needs-squash
此处折叠了201条消息 查看更多
CANN-robot
CANN-robot成员
8月3日 评论:

The MR can not be merged, because of CodeReview discussion not resolved

If you want to solve this problem, you can click here to do it in the FAQs.

likedislike
TangPCTangPC成员
8月3日 解决了最后一个问题
CANN-robotCANN-robot成员
8月3日 关闭了关联的issue
CANN-robotCANN-robot成员
8月3日 合入了pull request
CANN-robot
CANN-robot成员
8月3日 评论:

Pull Request 已合并或已关闭。

If you want to solve this problem, you can click here to do it in the FAQs.

likedislike