Star0

Star0

cann-robotfeat(aicpu): enable constant folding for migrated ops

e5540cc9创建于 26 天前历史提交

文件	最后提交记录	最后更新时间
docs	fix: 补齐激活算子dtype文档并修复Quantize错误日志格式 Co-authored-by: jiaoyiming<jiaoyiming2@huawei.com> # message auto-generated for no-merge-commit merge: !10091 merge fix/b-type-dtype-compliance into master fix: 补齐激活算子dtype文档并修复Quantize错误日志格式 Created-by: jiaoyiming Commit-by: jiaoyiming Merged-by: cann-robot Description: ## 描述部分激活算子的接口文档遗漏已有支持类型，与 API 错误日志列表不一致；Quantize 的支持列表重复包裹方括号，影响类型识别。 - Elu、LeakyRelu 的 self/out 文档补齐 DOUBLE。 - Relu 的 self/out 文档补齐 UINT8，并同步 Elu、Relu 的 Atlas 训练系列说明。 - Sigmoid 的 self 文档补齐 DOUBLE、INT8、INT16、INT32、INT64、UINT8、BOOL、COMPLEX64、COMPLEX128；out 文档补齐 DOUBLE、COMPLEX64、COMPLEX128。 - Quantize 将 `support list [%s]` 改为 `support list %s`，避免 ToString 自带方括号导致 `[[...]]`。本次不改变类型校验范围或计算逻辑。Quantize 当前文档已包含 zeroPoints 的 FLOAT32 及输出 FLOAT8 类型，无需重复修改。 ## 关联的Issue Fixes #5688 https://gitcode.com/cann/ops-nn/issues/5688 ## 测试 - git diff --check 通过。 - 对照 API 类型列表静态核对相关文档，确认一致。 - 检查 Quantize 日志格式只保留类型列表自身的一层方括号。 - 当前环境未执行 CANN 编译、UT、二级冒烟或设备日志复测，需由后续 CI/设备环境验证。 ## 文档更新更新 activation/elu、activation/leaky_relu、activation/relu、activation/sigmoid 下对应的 aclnn 接口文档。 ## 类型标签 - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他 ## AI/Agent生成声明 - [x] AI辅助编写 See merge request: cann/ops-nn!10091	1 个月前
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	3 个月前
framework	fix: migrate simple onnx parse params plugins Co-authored-by: clinglai0517<laichangling@huawei.com> # message auto-generated for no-merge-commit merge: !9161 merge onnx-fix-0825 into master fix: migrate simple onnx parse params plugins Created-by: clinglai0517 Commit-by: clinglai0517 Merged-by: cann-robot Description: ## 描述将一批 ONNX 插件的参数解析接口从 ParseParamsFn 迁移到 ParseParamsByOperatorFn，实现onnx插件和protobuf的解耦涉及算子： - Mish - FastGelu - Size - Elu - LeakyRelu - HardMax 【实现说明】 - 无属性或仅默认属性的算子，直接切换新注册接口。 - 对 Elu、LeakyRelu、HardMax，从 ge::Operator 的 attribute JSON 中解析 ONNX 属性。 - 保留旧逻辑中的默认值行为。 - 对缺失或非数组形式的 attribute JSON 增加保护，避免无属性场景误返回 FAILED。 ## 关联的Issue Issue #5115 ## 测试 onnx插件ut编译、执行通过 ## 文档更新不涉及 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [x] AI辅助编写 See merge request: cann/ops-nn!9161	1 个月前
op_api	清理stub下的重复头文件 Co-authored-by: yue-ma<mayue54@huawei.com> # message auto-generated for no-merge-commit merge: !9219 merge master into master 清理stub下的重复头文件 Created-by: yue-ma Commit-by: yue-ma Merged-by: cann-robot Description: ## 描述本 PR 主要聚焦于清理 common/stub/op_api/level0/ 目录下与各模块 common/op_api/ 目录重复的算子声明头文件：删除 adaptive_avg_pool2d_assist_matrix.h、batch_norm_backward.h、gather_elements.h、index_put.h、max_pool3d_with_argmax_v2.h、threshold.h、unique_with_counts_and_sorting.h 等约 27 个头文件，并将受影响 .cpp 文件中的 #include 从 level0/xxx.h 统一改为指向各模块的规范头文件路径，消除头文件的重复维护。整体属于代码整理类改动，不涉及对外接口和业务行为的变化。 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> https://gitcode.com/cann/ops-nn/issues/5159 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> UT/ST通过 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> 不涉及 ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!9219	1 个月前
op_graph	补齐A5算子op_graph下面的cmakelist、同步修改apply_adam_w算子geir通路动态用例失败问题 Co-authored-by: t00850857<tanxin44@huawei.com> # message auto-generated for no-merge-commit merge: !714 merge master into master 补齐A5算子op_graph下面的cmakelist、同步修改apply_adam_w算子geir通路动态用例失败问题 Created-by: tan_xin Commit-by: t00850857 Merged-by: cann-robot Description: ## 描述 1、补齐A5elu、gelu、gather_v2等算子op_graph下面的cmakelist 2、同步修改apply_adam_w算子geir通路动态用例失败问题 ## 关联的Issue [[DTS2025122942281](https://dts-szv.clouddragon.huawei.com/DTSPortal/ticket/DTS2025122942281)] ApplyAdamW geir通路功能有用例静态成功动态失败 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!714	8 个月前
op_host	feat: 增加 ascend350 平台支持（25 个激活/量化算子）并重构 SocVersion 判断为 NpuArch Co-authored-by: wenqi_wu<wenqiwu0628@163.com> # message auto-generated for no-merge-commit merge: !9834 merge support-ascend350-activation-ops into master feat: 增加 ascend350 平台支持（25 个激活/量化算子）并重构 SocVersion 判断为 NpuArch Created-by: wenqi_wu Commit-by: wenqi_wu Merged-by: cann-robot Description: ## 描述为以下 25 个算子增加 ascend350（arch35）平台支持，并将 7 个 aclnn 接口的 SocVersion 判断重构为 NpuArch 判断。算子列表（ascend350 支持，25 个）： - 量化类：AscendQuant / AscendAntiQuant / Quantize（quant） - Elu / EluGrad / EluGradV2（activation） - FastGelu / FastGeluGrad - Gelu / GeluGrad / GeluV2 / GeluGradV2 - ReluGrad / ReluGradV2 / ReluV2 - LeakyReluGrad - SigmoidGrad / SiluGrad - Swish / SwishGrad - HardtanhGrad - Mish / MishGrad - LogSigmoid / LogSigmoidGrad 注：LeakyRelu 的 ascend350 支持已由上游先行合入，本 PR 不含其重复改动。改动内容： 1. 算子 CMakeLists.txt：SUPPORT_COMPUTE_UNIT 追加 ascend350，SUPPORT_TILING_DIR 追加对应 arch35； 2. op_host/_def.cpp：为每个算子追加 this->AICore().AddConfig("ascend350", ...)； 3. 新增 op_host/config/ascend350/_binary.json 二进制配置文件； 4. scripts/kernel/binary_config/ascendc_config.json：为上述算子增加 ascend350 的 compute_units 与 compile_options（AscendAntiQuant 上游本无条目，走默认编译选项）； 5. op_api 中 SocVersion → NpuArch 重构（7 处）： - aclnn_rrelu_with_noise.cpp：GenerateUniformRandom 中平台判断改为 GetCurNpuArch() == NpuArch::DAV_2201（语义等价，910B/910_93 均对应 DAV_2201）； - aclnn_sigmoid_backward.cpp / aclnn_swish_backward.cpp / aclnn_gelu_v2.cpp：CheckSocVersionIsSupportBf16 改为 curArch == NpuArch::DAV_2201 \|\| IsRegbase(curArch)； - aclnn_sigmoid.cpp：CheckSocVersionIsSupportBf16 与 GetSelfRefDtypeList 改为 curArch == NpuArch::DAV_2201 \|\| IsRegbase(curArch)； - aclnn_binary_cross_entropy_with_logits_target_backward.cpp：GetDtypeSupportList 改为 NpuArch 判断； - aclnn_logsigmoid_backward.cpp：GetInputDtypeSupportList / GetOutputDtypeSupportList 改为 NpuArch 判断。 ## 关联的Issue https://gitcode.com/cann/ops-nn/issues/5538 ## 测试 <!--TODO--> ## 文档更新无需更新文档 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其它（请说明） ## AI/Agent辅助编译 - [x] AI辅助编写 See merge request: cann/ops-nn!9834	1 个月前
op_kernel	MicroAPI namespace to Reg for arch35 kernels Co-authored-by: gcw_DS4cmz2b<1312925094@qq.com> # message auto-generated for no-merge-commit merge: !9267 merge rename into master MicroAPI namespace to Reg for arch35 kernels Created-by: gcw_DS4cmz2b Commit-by: gcw_DS4cmz2b Merged-by: cann-robot Description: ## 描述修改了index,activation,matmul,conv,foreach,optim,experimental,vfusion,hash这几个文件夹内的命名空间，将MicroAPI-->Reg ## 关联的Issue [多个文件夹内的算子命名空间使用的MicroAPI，应改为Reg](https://gitcode.com/cann/ops-nn/issues/5218) ## 测试验证方法严格按四步流程，逐算子验证： 1. Step 1：编译原始代码，保存 baseline binary .o 文件 MD5 2. Step 2：执行 MicroAPI → Reg 替换 3. Step 3：编译替换后代码，保存 after binary .o 文件 MD5 4. Step 4：对比 A_before.md5 与 B_after.md5 平台：ascend950，编译参数：--soc=ascend950 --ops=<op> -j16 验证替换前后 binary 产物 MD5 完全一致。冒烟，代码审查均通过 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!9267	1 个月前
op_kernel_aicpu	feat(aicpu): enable constant folding for migrated ops Co-authored-by: hid29204727<wangboyu20@huawei.com> # message auto-generated for no-merge-commit merge: !9280 merge codex/aicpu-const-folding into master feat(aicpu): enable constant folding for migrated ops Created-by: hid29204727 Commit-by: hid29204727 Merged-by: cann-robot Description: ## 描述补充 ops-nn 仓 AICPU host 常量折叠注册通路，并为当前已标记 `OPEN_OPS_FLAG` 的 18 个迁移算子统一开启常量折叠支持。主要改动： - 新增仓级 `OPS_NN_REGISTER_CPU_KERNELV2` 注册包装宏，host 构建优先注册 V2，device/custom/UT 保持 V1 回退。 - 为 `add_modules_sources` 和 `add_all_modules_sources` 增加 `HOSTCPU` 参数及 host OBJECT 收集逻辑。 - 为 18 个 `OPEN_OPS_FLAG` 算子补充 `HOSTCPU TRUE`、注册头文件和 V2 注册。 - host OBJECT 汇入 `libopconstant_folding_nn.so`。 ## 关联的Issue 无。 ## 测试已完成以下静态验证： - `git diff --check` 通过。 - 18 个 `OPEN_OPS_FLAG` 算子均具备 `HOSTCPU TRUE`。 - 18 个算子均使用 `OPS_NN_REGISTER_CPU_KERNELV2`，无直接 V1 注册残留。 - 注册头文件、host 编译宏和仓级 so 收集通路检查通过。当前 Windows 环境缺少 CANN、NPU、CMake 和远端构建桥，未执行 `bash build.sh --pkg -j16`；完整打包构建及常量折叠功能验证待 CANN Linux 环境/CI 完成。 ## 文档更新无仓内文档变更。 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 - [x] AI辅助编写 See merge request: cann/ops-nn!9280	26 天前
tests	feat: Add aclnn golden functions and test cases Co-authored-by: yanzhi2024<yanzhi@huawei.com> # message auto-generated for no-merge-commit merge: !9487 merge master into master feat: Add aclnn golden functions and test cases Created-by: yanzhi2024 Commit-by: yanzhi2024 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 添加蓝区aclnn用例和golden函数 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> https://gitcode.com/cann/ops-nn/issues/5325 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> 不涉及 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> 不涉及 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [x] AI辅助编写 See merge request: cann/ops-nn!9487	1 个月前
CMakeLists.txt	feat(aicpu): enable constant folding for migrated ops Co-authored-by: hid29204727<wangboyu20@huawei.com> # message auto-generated for no-merge-commit merge: !9280 merge codex/aicpu-const-folding into master feat(aicpu): enable constant folding for migrated ops Created-by: hid29204727 Commit-by: hid29204727 Merged-by: cann-robot Description: ## 描述补充 ops-nn 仓 AICPU host 常量折叠注册通路，并为当前已标记 `OPEN_OPS_FLAG` 的 18 个迁移算子统一开启常量折叠支持。主要改动： - 新增仓级 `OPS_NN_REGISTER_CPU_KERNELV2` 注册包装宏，host 构建优先注册 V2，device/custom/UT 保持 V1 回退。 - 为 `add_modules_sources` 和 `add_all_modules_sources` 增加 `HOSTCPU` 参数及 host OBJECT 收集逻辑。 - 为 18 个 `OPEN_OPS_FLAG` 算子补充 `HOSTCPU TRUE`、注册头文件和 V2 注册。 - host OBJECT 汇入 `libopconstant_folding_nn.so`。 ## 关联的Issue 无。 ## 测试已完成以下静态验证： - `git diff --check` 通过。 - 18 个 `OPEN_OPS_FLAG` 算子均具备 `HOSTCPU TRUE`。 - 18 个算子均使用 `OPS_NN_REGISTER_CPU_KERNELV2`，无直接 V1 注册残留。 - 注册头文件、host 编译宏和仓级 so 收集通路检查通过。当前 Windows 环境缺少 CANN、NPU、CMake 和远端构建桥，未执行 `bash build.sh --pkg -j16`；完整打包构建及常量折叠功能验证待 CANN Linux 环境/CI 完成。 ## 文档更新无仓内文档变更。 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 - [x] AI辅助编写 See merge request: cann/ops-nn!9280	26 天前
README.md	Label the chip Co-authored-by: zhouwenfang<zhouwenfang3@h-partners.com> # message auto-generated for no-merge-commit merge: !7823 merge master into master Label the chip Created-by: zhouwenfang Commit-by: zhouwenfang Merged-by: cann-robot Description: ## 描述 Label the chip ## 关联的Issue https://gitcode.com/cann/ops-nn/issues/4245 ## 测试 NA ## 文档更新 activation目录所有的aclnn.md文件 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!7823	2 个月前

Elu

产品支持情况

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

功能说明

接口功能：对输入张量self中的每个元素x调用指数线性单元激活函数ELU，并将得到的结果存入输出张量out中。
计算公式：
$\begin{cases} scale \ast x, \quad x > 0\\ \alpha \ast scale \ast (exp(x \ast inputScale)-1), \quad x \leq 0 \end{cases}$

参数说明：

参数名	输入/输出	描述	使用说明	数据类型	数据格式	维度(shape)	非连续Tensor
self	输入	表示ELU激活函数的输入，公式中的x。	支持空Tensor。	FLOAT、FLOAT16、BFLOAT16	ND	0-8	√
alpha	输入	表示ELU激活函数的激活系数，公式中的α。	数据类型需要是可转换为FLOAT的数据类型。	-	-	-	-
scale	输入	表示ELU激活函数的缩放系数，公式中的scale。	数据类型需要是可转换为FLOAT的数据类型。	-	-	-	-
inputScale	输入	表示ELU激活函数的输入的缩放系数，公式中的inputScale。	数据类型需要是可转换为FLOAT的数据类型。	-	-	-	-
out	输出	表示ELU激活函数的输出。	shape需要与self一致。	FLOAT、FLOAT16、BFLOAT16	ND	0-8	√

Atlas 训练系列产品：数据类型支持FLOAT、FLOAT16。

约束说明

无

调用说明

调用方式	调用样例	说明
aclnn调用	test_aclnn_elu	通过aclnnElu&aclnnInplaceElu接口方式调用Elu算子。
aclnn调用	test_aclnn_inplace_elu	通过aclnnElu&aclnnInplaceElu接口方式调用Elu算子。