| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
新增ApplyMomentum算子 Co-authored-by: cheng-ziyang2<chengziyang2@huawei.com> # message auto-generated for no-merge-commit merge: !1403 merge ApplyMomentum into master 新增ApplyMomentum算子 Created-by: cheng-ziyang2 Commit-by: cheng-ziyang2 Merged-by: cann-robot Description: ## 描述 问题/功能描述 本次PR新增了ApplyMomentum的优化器算子，实现了基于动量的参数更新算法。该算子支持标准动量和Nesterov动量两种更新方式，并兼容多种浮点数据类型（如float16, float32, bfloat16），用于在模型训练过程中高效地更新模型参数和累积梯度。 修改方案描述 本次提交完整实现了ApplyMomentum算子的全栈功能。具体包括：1) 算子定义、属性注册、形状/类型推断及构建系统集成；2) AscendC内核实现，包含核心计算逻辑、数据流图定义及Tiling策略；3) 针对Ascend 950平台的编译配置与二进制生成规则；4) 新增了Tiling和InferShape功能的单元测试，以验证算子逻辑的正确性。整个实现覆盖了从框架接口到硬件内核的完整链路。 ## 关联的Issue 关联Issue [#918](https://gitcode.com/cann/ops-nn/issues/918) ## 测试 泛化测试通过 ## 文档更新 不涉及 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!1403 | 5 个月前 | |
新增ApplyMomentum算子 Co-authored-by: cheng-ziyang2<chengziyang2@huawei.com> # message auto-generated for no-merge-commit merge: !1403 merge ApplyMomentum into master 新增ApplyMomentum算子 Created-by: cheng-ziyang2 Commit-by: cheng-ziyang2 Merged-by: cann-robot Description: ## 描述 问题/功能描述 本次PR新增了ApplyMomentum的优化器算子，实现了基于动量的参数更新算法。该算子支持标准动量和Nesterov动量两种更新方式，并兼容多种浮点数据类型（如float16, float32, bfloat16），用于在模型训练过程中高效地更新模型参数和累积梯度。 修改方案描述 本次提交完整实现了ApplyMomentum算子的全栈功能。具体包括：1) 算子定义、属性注册、形状/类型推断及构建系统集成；2) AscendC内核实现，包含核心计算逻辑、数据流图定义及Tiling策略；3) 针对Ascend 950平台的编译配置与二进制生成规则；4) 新增了Tiling和InferShape功能的单元测试，以验证算子逻辑的正确性。整个实现覆盖了从框架接口到硬件内核的完整链路。 ## 关联的Issue 关联Issue [#918](https://gitcode.com/cann/ops-nn/issues/918) ## 测试 泛化测试通过 ## 文档更新 不涉及 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!1403 | 5 个月前 |
| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
| 5 个月前 | ||
| 5 个月前 |