Star0

Star0

cann-robotfeat: 增加 ascend350 平台支持（25 个激活/量化算子）并重构 SocVersion 判断为 NpuArch

a84fd19a创建于 9月12日历史提交

文件	最后提交记录	最后更新时间
docs	修复算子文档格式与示例代码规范问题 Co-authored-by: yuanbin_22<yuanbin22@huawei.com> # message auto-generated for no-merge-commit merge: !8288 merge fix-doc-format-issues into master 修复算子文档格式与示例代码规范问题 Created-by: yuanbin_22 Commit-by: yuanbin_22 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> fix: 修复算子文档格式与示例代码规范问题 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> https://gitcode.com/cann/ops-nn/issues/4603 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [x] AI辅助编写 See merge request: cann/ops-nn!8288	2 个月前
examples	修复算子文档格式与示例代码规范问题 Co-authored-by: yuanbin_22<yuanbin22@huawei.com> # message auto-generated for no-merge-commit merge: !8288 merge fix-doc-format-issues into master 修复算子文档格式与示例代码规范问题 Created-by: yuanbin_22 Commit-by: yuanbin_22 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> fix: 修复算子文档格式与示例代码规范问题 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> https://gitcode.com/cann/ops-nn/issues/4603 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [x] AI辅助编写 See merge request: cann/ops-nn!8288	2 个月前
framework	TF plugin开源 Co-authored-by: sunchun<sunchun4@h-partners.com> # message auto-generated for no-merge-commit merge: !7516 merge nn into master TF plugin开源 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!7516	3 个月前
op_api	nn仓算子错误日志整改 Co-authored-by: qq_52874943<gaomingyuan3@h-partners.com> # message auto-generated for no-merge-commit merge: !9792 merge master into master nn仓算子错误日志整改 Created-by: qq_52874943 Commit-by: qq_52874943 Merged-by: cann-robot Description: ## 描述该 PR 围绕 nn 仓中 elu/gelu/geglu 系列算子的错误日志整改展开：将 test_geir_elu_backward.cpp、test_geir_gelu_quant.cpp 等示例中初始化、RunGraph、GEFinalize 失败路径的打印级别由 INFO 提升为 ERROR，同时批量修正 CheckShape、CheckFormat、CheckDtypeValid、GetPlatformInfo、InferShapeForGeGluGradV2 等函数中 OP_LOGE/OP_LOGW 日志的拼写与语法错误，并顺带调整了 aclnn_gelu_quant.cpp 中一处形状校验条件及其提示文案，整体不涉及接口、数据结构的变更，主要提升错误日志的准确性、语义一致性与可读性。 ## 关联的Issue 关联Issue[#5183](https://gitcode.com/cann/ops-nn/issues/5183) ## 测试本地已完成新精度对比与性能测试，通过静态检查、蓝区冒烟、算子ST以及算子二级冒烟 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!9792	1 个月前
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	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 个月前
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: 增加 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 个月前
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	3 个月前

Gelu

产品支持情况

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

功能说明

算子功能：高斯误差线性单元激活函数。
计算公式：
$\sqrt{2 / \pi} * (self + 0.044715 * self^{3})))$

参数说明

参数名	输入/输出	描述	使用说明	数据类型	数据格式	维度(shape)	非连续Tensor
x	输入	表示GELU激活函数的输入，公式中的self。	数据类型必须和out一样。 shape必须和out一样。支持空Tensor。	FLOAT、FLOAT16、BFLOAT16	ND	0-8	√
y	输出	表示GELU激活函数的输出。	数据类型必须和self一致。 shape必须和self一致。	FLOAT、FLOAT16、BFLOAT16	ND	0-8	√

Atlas 200I/500 A2 推理产品、Atlas 推理系列产品、Atlas 训练系列产品：数据类型支持FLOAT、FLOAT16。

约束说明

无

调用说明

调用方式	样例代码	说明
aclnn接口	test_aclnn_gelu.cpp	通过aclnnGelu接口方式调用Gelu算子。