Star0

Star0

文件	最后提交记录	最后更新时间
docs	产品名称修改 Co-authored-by: y60124828<yanglou1@h-partners.com> # message auto-generated for no-merge-commit merge: !11292 merge master into master 产品名称修改 Created-by: yanglu-1 Commit-by: y60124828 Merged-by: cann-robot Description: ## 描述产品名称修改 ## 关联的Issue 关联Issue [#6089](https://gitcode.com/cann/ops-nn/issues/6089) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!11292	22 天前
examples	[fix] 补充examples目录下10个示例代码文件缺失的版权声明头 Co-authored-by: tangweiwei<tangweiwei2@huawei.com> # message auto-generated for no-merge-commit merge: !10258 merge fix/example-copyright-header into master [fix] 补充examples目录下10个示例代码文件缺失的版权声明头 Created-by: tangweiwei2 Commit-by: tangweiwei Merged-by: cann-robot Description: ## 描述检查仓内所有 .cpp/.cc/.h 源文件（排除 experimental 目录）的版权声明情况，发现 examples 目录下 10 个示例代码文件缺失标准版权声明头（Copyright (c) 20xx Huawei Technologies Co., Ltd. ... CANN Open Software License Agreement Version 2.0 ...）。本 PR 为以下 10 个文件补充版权声明头，版权年份与各算子同目录下其他源文件保持一致： 1. examples/add_example/examples/test_geir_add_example.cpp（2025） 2. index/scatter_mul/examples/test_geir_scatter_mul.cpp（2026） 3. index/scatter_max/examples/test_geir_scatter_max.cpp（2026） 4. index/scatter_min/examples/test_geir_scatter_min.cpp（2026） 5. index/scatter_div/examples/test_geir_scatter_div.cpp（2026） 6. activation/swiglu_group/examples/test_aclnn_swiglu_group.cpp（2026） 7. vfusion/modulate_grad/examples/test_aclnn_modulatebackward.cpp（2025） 8. quant/swiglu_mx_quant/examples/arch35/test_aclnn_swiglu_mx_quant.cpp（2026） 9. quant/swiglu_mx_quant_with_dual_axis/examples/arch35/test_aclnn_swiglu_mx_quant_with_dual_axis.cpp（2026） 10. quant/swiglu_backward_mx_quant_with_dual_axis/examples/arch35/test_aclnn_swiglu_backward_mx_quant_with_dual_axis.cpp（2026）仅在每个文件头部新增版权声明，不改动任何功能代码。 ## 关联的Issue 关联Issue #5782 ## 测试 - 使用脚本扫描确认修改前 10 个文件无版权声明，修改后均包含标准版权头； - 仅新增文件头部注释，不影响编译与功能，无需运行算子测试。 ## 文档更新无 ## 类型标签 - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 - [x] AI辅助编写 See merge request: cann/ops-nn!10258	1 个月前
framework	950系列支持 scatter_max/min/mul/div、scatter_list、foreach addcmul/addcdiv_list Co-authored-by: tangpingchuan<tangpingchuan@huawei.com> # message auto-generated for no-merge-commit merge: !5991 merge master into master 950系列支持 scatter_max/min/mul/div、scatter_list、foreach addcmul/addcdiv_list Created-by: zl_hw Commit-by: tangpingchuan Merged-by: cann-robot Description: ## 描述支持950系列 scatterXXX 长尾算子；支持ForeachXXXList 长尾算子 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> ## 测试 ut/st 所有host ut全覆盖，且覆盖率全达标；所有算子ttk泛化100+用例全部Pass ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!5991	4 个月前
op_graph	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	3 个月前
op_host	fix(arch35): scatter 类补 AutoContiguous；LambApplyOptimizerAssign infershape 与 tiling 对齐；foreach inplace 与 ScatterList 资料订正 Co-authored-by: tangpingchuan<tangpingchuan@huawei.com> # message auto-generated for no-merge-commit merge: !8767 merge fix/scatter-noncontig-and-lamb-infershape into master fix(arch35): scatter 类补 AutoContiguous；LambApplyOptimizerAssign infershape 与 tiling 对齐；foreach inplace 与 ScatterList 资料订正 Created-by: zl_hw Commit-by: tangpingchuan Merged-by: cann-robot Description: ## 描述长尾算子问题优化：scatter 类补 AutoContiguous；LambApplyOptimizerAssign infershape 与 tiling 对齐；foreach inplace 与 ScatterList 资料订正 ## 关联的Issue https://gitcode.com/cann/ops-nn/issues/4865 ## 测试不涉及 ## 文档更新 README ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [x] AI辅助编写 See merge request: cann/ops-nn!8767	2 个月前
op_kernel	950系列支持 scatter_max/min/mul/div、scatter_list、foreach addcmul/addcdiv_list Co-authored-by: tangpingchuan<tangpingchuan@huawei.com> # message auto-generated for no-merge-commit merge: !5991 merge master into master 950系列支持 scatter_max/min/mul/div、scatter_list、foreach addcmul/addcdiv_list Created-by: zl_hw Commit-by: tangpingchuan Merged-by: cann-robot Description: ## 描述支持950系列 scatterXXX 长尾算子；支持ForeachXXXList 长尾算子 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> ## 测试 ut/st 所有host ut全覆盖，且覆盖率全达标；所有算子ttk泛化100+用例全部Pass ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!5991	4 个月前
tests	docs(index): scatter Mul/Div/Max/Min 资料与 UT 注释同步 int64 直排后的支持面 Co-authored-by: tangpingchuan<tangpingchuan@huawei.com> # message auto-generated for no-merge-commit merge: !10794 merge fix/scatter-doc-ut-sync into master docs(index): scatter Mul/Div/Max/Min 资料与 UT 注释同步 int64 直排后的支持面 Created-by: zl_hw Commit-by: tangpingchuan Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!10794	1 个月前
CMakeLists.txt	950系列支持 scatter_max/min/mul/div、scatter_list、foreach addcmul/addcdiv_list Co-authored-by: tangpingchuan<tangpingchuan@huawei.com> # message auto-generated for no-merge-commit merge: !5991 merge master into master 950系列支持 scatter_max/min/mul/div、scatter_list、foreach addcmul/addcdiv_list Created-by: zl_hw Commit-by: tangpingchuan Merged-by: cann-robot Description: ## 描述支持950系列 scatterXXX 长尾算子；支持ForeachXXXList 长尾算子 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> ## 测试 ut/st 所有host ut全覆盖，且覆盖率全达标；所有算子ttk泛化100+用例全部Pass ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!5991	4 个月前
README.md	docs(scatter_max/min): 按 A2 图模式实测补充芯片维度数据类型说明 Co-authored-by: bartonfang<fangziyang1@huawei.com> # message auto-generated for no-merge-commit merge: !11154 merge docs-scatter-dtype-a2-a5 into master docs(scatter_max/min): 按 A2 图模式实测补充芯片维度数据类型说明 Created-by: bartonfang Commit-by: bartonfang Merged-by: cann-robot Description: ## 背景 PR #9223 为 scatter_max/min/mul/div、scatter_nd_sub 等 README 的“产品支持情况”补充了 A2/A3 标记，但参数说明中的数据类型只有一套（A5/950 首次适配时写入的 FLOAT16/FLOAT32/INT32/INT8/UINT8），未按芯片区分。本次按图模式实测确认 A2/A3 的真实支持范围后修正文档。 ## 实测方法在 A2（910B3，CANN 25.5.1，device 6）上使用图模式：TF GraphDef（对应 TF 的 ScatterMax/Min/Div/Mul/NdSub 算子）→ `atc --framework=3 --soc=Ascend910B3` 编译 om → `aclmdl` 加载执行，输出与 CPU 参考对拍。ONNX 可表达的 ScatterElements/ScatterND 用 ONNX→ATC 同样跑通。 ## 实测结果（A2，图模式，✅=编译+执行+对拍通过，✗=ATC/GE 编译失败） \| 算子 \| FLOAT16 \| FLOAT32 \| INT32 \| INT64 \| DOUBLE \| INT8 \| UINT8 \| INT16 \| BOOL \| \|---\|---\|---\|---\|---\|---\|---\|---\|---\|---\| \| ScatterMax \| ✅ \| ✅ \| ✅ \| ✅ \| ✅ \| ✗ \| ✗ \| ✗ \| ✗ \| \| ScatterMin \| ✅ \| ✅ \| ✅ \| ✅ \| ✅ \| ✗ \| ✗ \| ✗ \| ✗ \| \| ScatterDiv \| ✅ \| ✅ \| ✅ \| ✗ \| ✗ \| ✅ \| ✅ \| ✗ \| ✗ \| \| ScatterMul \| ✅ \| ✅ \| ✅ \| ✗ \| ✗ \| ✅ \| ✅ \| ✗ \| ✗ \| \| ScatterNdSub \| ✅ \| ✅ \| ✅ \| ✗ \| ✗ \| ✅ \| ✅ \| ✗ \| ✗ \| \| ScatterElements(ONNX图) \| ✅ \| ✅ \| ✅ \| ✅ \| ✅ \| ✅ \| ✅ \| ✅ \| ✅ \| \| ScatterND(ONNX图) \| ✅ \| ✅ \| ✅ \| 待补 \| ✅ \| ✅ \| ✅ \| ✅ \| ✅ \| （ScatterDiv 的 FLOAT32 在 ATC 编译与执行均成功，与 CPU 参考的差异来自浮点精度，非不支持。） ## 本次改动（2 个文件） - index/scatter_max/README.md - index/scatter_min/README.md 在参数说明表格后补充芯片维度说明： - Ascend 950PR/Ascend 950DT：支持 FLOAT16、FLOAT32、INT32、INT8、UINT8； - Atlas A2/A3：支持 FLOAT16、FLOAT32、INT32、INT64、DOUBLE，不支持 INT8、UINT8。 ## 其它算子的处理 - scatter_div / scatter_mul / scatter_nd_sub：实测支持 FLOAT16/FLOAT32/INT32/INT8/UINT8，与 README 表格一致，因此撤销了先前基于发布配置加的 A2/A3 限制说明，本次不再改动这三个文件。 - scatter_add/sub、scatter_nd_add/max/min/update、scatter_elements(_v2) 等：README 为 aclnn 接口支持范围（含 AICPU/其它实现兜底），与 kernel 发布列表不是同一口径，建议后续按同样方法逐个实测后再决定是否补充说明。 See merge request: cann/ops-nn!11154	12 天前

ScatterMax

产品支持情况

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

功能说明

算子功能：实现兼容tf.compat.v1.scatter_max的功能，将tensor updates中的值按指定的索引tensor indices逐元素取最大值更新到tensor var的切片上。若有多于一个updates值作用到var的同一个切片，则依次在该切片上取最大值。属于原地（in-place）更新，输出复用输入var。规则如下：
$\max(var[indices[i], ...], updates[i, ...])$

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
var	输入	支持空Tensor。表示待被更新的张量，Device侧的aclTensor，原地更新。shape支持1维及以上，数据类型需与updates一致。	FLOAT16、FLOAT32、INT32、INT8、UINT8	ND
indices	输入	支持空Tensor。表示待更新的索引张量，Device侧的aclTensor。shape支持1维及以上，数据类型为INT32或INT64。支持索引越界，越界索引（小于0或大于等于var第一维）会被静默跳过，不做更新。	INT32、INT64	ND
updates	输入	支持空Tensor。表示需要与var逐元素取最大值的张量，Device侧的aclTensor。shape支持1维及以上，需满足updates.shape = indices.shape + var.shape[1:]，数据类型需与var一致。	FLOAT16、FLOAT32、INT32、INT8、UINT8	ND
var	输出	支持空Tensor。表示更新后的张量，Device侧的aclTensor，与输入var共享内存。shape和数据类型均与输入var一致。	FLOAT16、FLOAT32、INT32、INT8、UINT8	ND
use_locking	属性	HOST侧的bool值，是否对更新加锁，默认false。	bool	-

Ascend 950PR&950DT系列产品：var、updates的数据类型支持FLOAT16、FLOAT32、INT32、INT8、UINT8。
Atlas A3系列产品、Atlas A2系列产品：var、updates的数据类型支持FLOAT16、FLOAT32、INT32、INT64、DOUBLE，不支持INT8、UINT8。

约束说明

确定性计算：默认确定性实现。

调用说明

调用方式	样例代码	说明
aclnn接口	test_aclnn_scatter_max	通过aclnnScatterMax接口方式调用ScatterMax算子。