cann-robotmodify SparseApplyAdagradV2 updateSlots

文件	最后提交记录	最后更新时间
examples	format cpp Co-authored-by: yang-di52<yangdi52@huawei.com> # message auto-generated for no-merge-commit merge: !6784 merge issue_fix into master format cpp Created-by: yang-di52 Commit-by: yang-di52 Merged-by: cann-robot Description: ## 描述批量刷新cpp代码格式 ## 关联的Issue [#3791](https://gitcode.com/cann/ops-nn/issues/3791) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述：代码格式化 ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!6784	17 天前
framework	TF plugin 迁移2 Co-authored-by: sunchun<sunchun4@h-partners.com> # message auto-generated for no-merge-commit merge: !7586 merge nn2 into master TF plugin 迁移2 Created-by: sunchun Commit-by: sunchun Merged-by: cann-robot Description: ## 描述 TF plugin 迁移 ## 关联的Issue https://gitcode.com/cann/ops-nn/issues/4196 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!7586	1 天前
op_graph	feat(sparse_apply_adagrad_v2 ): add AscendC sparse_apply_adagrad_v2 operator implementation Co-authored-by: rsj007<renshuojian@huawei.com> # message auto-generated for no-merge-commit merge: !6425 merge master into master feat(sparse_apply_adagrad_v2 ): add AscendC sparse_apply_adagrad_v2 operator implementation Created-by: rsj007 Commit-by: rsj007 Merged-by: cann-robot Description: ## 描述 add AscendC sparse_apply_adagrad_v2 operator implementation ## 关联的Issue https://gitcode.com/cann/ops-nn/issues/3590 ## 测试黄区ttk测试ok ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!6425	23 天前
op_host	modify SparseApplyAdagradV2 updateSlots Co-authored-by: rsj007<renshuojian@huawei.com> # message auto-generated for no-merge-commit merge: !7789 merge master into master modify SparseApplyAdagradV2 updateSlots Created-by: rsj007 Commit-by: rsj007 Merged-by: cann-robot Description: ## 描述 1. 修改def里两个attr的顺序与proto一致； 2. readme里增加重复索引场景不保序的描述。 ## 关联的Issue https://gitcode.com/cann/ops-nn/issues/4244 ## 测试 geir测试、david、obp冒烟 ## 文档更新更新了README.md ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!7789	1 天前
op_kernel	sparse_apply_adagrad_v2 change to multi core Co-authored-by: rsj007<renshuojian@huawei.com> # message auto-generated for no-merge-commit merge: !7546 merge master into master sparse_apply_adagrad_v2 change to multi core Created-by: rsj007 Commit-by: rsj007 Merged-by: cann-robot Description: ## 描述 sparse_apply_adagrad_v2需要将分核改为多核策略 ## 关联的Issue https://gitcode.com/cann/ops-nn/issues/4156 ## 测试人工验证、冒烟 ## 文档更新无 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [x] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [x] AI辅助编写 See merge request: cann/ops-nn!7546	5 天前
tests	modify SparseApplyAdagradV2 updateSlots Co-authored-by: rsj007<renshuojian@huawei.com> # message auto-generated for no-merge-commit merge: !7789 merge master into master modify SparseApplyAdagradV2 updateSlots Created-by: rsj007 Commit-by: rsj007 Merged-by: cann-robot Description: ## 描述 1. 修改def里两个attr的顺序与proto一致； 2. readme里增加重复索引场景不保序的描述。 ## 关联的Issue https://gitcode.com/cann/ops-nn/issues/4244 ## 测试 geir测试、david、obp冒烟 ## 文档更新更新了README.md ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!7789	1 天前
CMakeLists.txt	feat(sparse_apply_adagrad_v2 ): add AscendC sparse_apply_adagrad_v2 operator implementation Co-authored-by: rsj007<renshuojian@huawei.com> # message auto-generated for no-merge-commit merge: !6425 merge master into master feat(sparse_apply_adagrad_v2 ): add AscendC sparse_apply_adagrad_v2 operator implementation Created-by: rsj007 Commit-by: rsj007 Merged-by: cann-robot Description: ## 描述 add AscendC sparse_apply_adagrad_v2 operator implementation ## 关联的Issue https://gitcode.com/cann/ops-nn/issues/3590 ## 测试黄区ttk测试ok ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!6425	23 天前
README.md	modify SparseApplyAdagradV2 updateSlots Co-authored-by: rsj007<renshuojian@huawei.com> # message auto-generated for no-merge-commit merge: !7789 merge master into master modify SparseApplyAdagradV2 updateSlots Created-by: rsj007 Commit-by: rsj007 Merged-by: cann-robot Description: ## 描述 1. 修改def里两个attr的顺序与proto一致； 2. readme里增加重复索引场景不保序的描述。 ## 关联的Issue https://gitcode.com/cann/ops-nn/issues/4244 ## 测试 geir测试、david、obp冒烟 ## 文档更新更新了README.md ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!7789	1 天前

SparseApplyAdagradV2

产品支持情况

产品	是否支持
Ascend 950PR/Ascend 950DT	√
Atlas A3 训练系列产品/Atlas A3 推理系列产品	×
Atlas A2 训练系列产品/Atlas A2 推理系列产品	×
Atlas 200I/500 A2 推理产品	×
Atlas 推理系列产品	×
Atlas 训练系列产品	×

功能说明

算子功能：SparseApplyAdagradV2是稀疏版本的AdagradV2优化器算子，根据稀疏梯度和索引向量原地更新参数变量和累加器。
计算公式：

update_slots=True)\text{accum}[\text{index}] \mathrel{+}= \text{grad}[i] \times \text{grad}[i] \quad (\text{when update\_slots=True})

var[index]−=lr×grad[i]accum[index]+ϵ\text{var}[\text{index}] \mathrel{-}= \text{lr} \times \frac{\text{grad}[i]}{\sqrt{\text{accum}[\text{index}]} + \epsilon}

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
var	输入	参数变量，shape (N, C)	FLOAT	ND
accum	输入	梯度平方累加器，shape (N, C)	FLOAT	ND
lr	输入	学习率标量	FLOAT	ND
epsilon	输入	常数因子标量，避免除零	FLOAT	ND
grad	输入	稀疏梯度，shape (M, C)	FLOAT	ND
indices	输入	稀疏索引向量，shape (M)	INT32	ND
var	输出	更新后的参数变量，shape (N, C)	FLOAT	ND
accum	输出	更新后的累加器，shape (N, C)	FLOAT	ND
use_locking	属性	是否在更新时加锁（NPU场景下通常忽略）	Bool	-
update_slots	属性	是否更新累加器（True时accum+=grad^2）	Bool	-

约束说明

var 和 accum 的 shape 必须相同
lr 和 epsilon 必须为标量（shape 为空或 [1]）
indices 必须为一维向量
grad 的第一维必须与 indices 的第一维相同，其余维度必须与 var 的对应维度相同
var、accum、lr、epsilon、grad 的 dtype 必须相同
indices 的 dtype 为 int32
越界索引（< 0 或 >= var 行数）将被跳过，不报错
当indices有重复索引值的时候，结果不保序

调用说明

调用方式	调用样例	说明
图模式调用	test_geir_sparse_apply_adagrad_v2	参见算子调用完成算子编译和验证。