Star0
0
代码
代码
Issues
Pull Requests
Actions
项目讨论
Wiki
分析
Star0
0
cann-robotcann-robot修正 Swish README 中 Atlas A2 的产品支持标记
ebfe4337创建于 21 天前历史提交
文件最后提交记录最后更新时间
docs
删除源码链接 Co-authored-by: fwx1169505<fengying27@h-partners.com> # message auto-generated for no-merge-commit merge: !12106 merge master into master 删除源码链接 Created-by: fengying555682 Commit-by: fwx1169505 Merged-by: cann-robot Description: ## 描述 删除源码链接 ## 关联的Issue [#6769](https://gitcode.com/cann/ops-nn/issues/6769) ## 测试 仅涉及文档修改，已经测试md文档语法 ## 文档更新 activation和quant目录下的md文档 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ x] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!12106
24 天前
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
3 个月前
op_api
refactor: 迁移NN公共op_api头文件引用 Co-authored-by: clinglai0517<laichangling@huawei.com> # message auto-generated for no-merge-commit merge: !10879 merge nn_prefix_common_headers_0920 into master refactor: 迁移NN公共op_api头文件引用 Created-by: clinglai0517 Commit-by: clinglai0517 Merged-by: cann-robot Description: ## 描述 本变更属于 NN 公共 op_api 头文件迁移的调用清理阶段： - 将非 experimental 目录中 21 个 op_api/level2_base.h 调用迁移到 op_api/level2_base_nn.h。 - 将非 experimental 目录中 2 个 op_api/level2_base_caculation.h 调用迁移到 op_api/level2_base_caculation_nn.h。 - experimental 目录暂时保留旧转发头：11 处 level2_base.h、3 处 level2_base_caculation.h，避免 CI 将导出同名 ACLNN 接口的实验算子共同链接。 - 保留旧头文件作为兼容转发头；本 MR 不宣称已完成最终包内 basename 去重。 - 不修改 op_api_def_nn.h，不包含功能逻辑变更。 ## 关联的Issue Issue #6008 ## 测试 - git diff --check upstream/master...HEAD：通过。 - pre_commit：全部通过，包括 clang-format、codespell 和 OAT。 - CANN 9.0 容器执行 bash build.sh --opapi -j32：通过，成功生成 libopapi_nn.so。 - 原 experimental CI 同时选择 swish 与 swish_v2 后发生既有同名 ACLNN 符号冲突；本次兼容调整已将两者移出最终 MR diff，等待线上 CI 复验。 ## 文档更新 无文档更新。 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述：公共头文件引用迁移 ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [x] AI辅助编写 See merge request: cann/ops-nn!10879
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
4 个月前
op_host
DynamicQuant等22个算子支持Ascend960dt Co-authored-by: TH_HW<taohai2@huawei.com> # message auto-generated for no-merge-commit merge: !11335 merge backBlue into master DynamicQuant等22个算子支持Ascend960dt Created-by: TH_HW Commit-by: TH_HW Merged-by: cann-robot Description: ## 描述 （+17812 / -2151）为一批 ops-nn 算子补齐 **Ascend 960DT（NPU Arch 9201）** 平台支持，并重构 GeGluV2 arch35 内核、为 Erfinv 增加 inplace 通路。 主要改动如下： 1. **Ascend960DT 平台适配（22 个算子）** - activation：Relu、ReluV2、ReluGrad、ReluGradV2、Sigmoid、SigmoidGrad、PRelu、PReluGradUpdate、SiluGrad、Softplus、Swish、SwishGrad、SwiGluGrad、GeGluV2。 - index：MapIndex、SparseSlice、QuantUpdateScatter。 - quant：Quantize、DynamicQuant、DynamicQuantV2、AntiMxQuant、GroupedDynamicMxQuant。 - 具体做法：CMakeLists.txt 的 SUPPORT_COMPUTE_UNIT / SUPPORT_TILING_DIR 增加 ascend960dt；*_def.cpp 增加 AddConfig("ascend960dt", ...)；aclnn/op_api 与 infershape 增加 NpuArch::DAV_9201 / Ascend960DT 分支；kernel 增加 __NPU_ARCH__ == 9201 分支；新增 op_host/config/ascend960dt/ 下的算子二进制配置（共 22 个，如 relu_binary.json、p_relu.json 等）；scripts/kernel/binary_config/ascendc_config.json 为相关算子登记 ascend960dt 计算单元与编译选项。 2. **GeGluV2 arch35 内核重构** - 将 arch35 模板按 erf / tanh 两类拆分，并补齐 align、align_last_axis_big、vreduce × fp16/bf16/fp32 变体，统一命名 *_arch35.h。 - tiling 新增 RegBase per-template per-element UB 预算模型（GetRegbasePerElem），并支持 ascend960dt。 - aclnnGeGlu 参数校验重构（CheckShape 支持 0D 输入与 dim 范围校验），新增 ST 用例（all_aclnnGeGlu(.V3).json）并调整 UT。 3. **Erfinv 支持 inplace**：新增 inplace dtype 支持列表与校验（GetInplaceDtypeSupportList / CheckInplaceDtypeValid），并补充 op_api UT。 4. **内核与接口修复** - SparseSlice：9201 路径改用 SetValue 写入 sliceStart/sliceEnd（替代 DataCopy），Reg::Reduce 使用完整命名空间限定。 - QuantUpdateScatter：9201 增加 FLOAT_OVERFLOW_MODE_CTRL 保护。 - ReluV2 infershape：Ascend960DT 不支持 NC1HWC0，UINT1 支持列表加入 Ascend960DT。 - DynamicQuant / Mx 量化：9201 上 fp8 scale 改为 bf16 乘法后再 cast 到 e8m0。 5. **代码规范**：统一 License 头注释格式、补齐文件末尾换行、调整 quant 系列 UT 的 CMakeLists；新增 MapIndex、SparseSlice 的 GE IR 图模式示例。 ## 关联的Issue 无。 ## 测试 - 新增 GeGluV2 ST 用例与 Erfinv inplace UT，补充/调整 GeGluV2 相关 UT。 - 新增 MapIndex / SparseSlice 的 GE IR 示例（examples/test_geir_*.cpp）。 - 通过 ascend960dt 编译与静态检查。 ## 文档更新 无。 ## 类型标签 <!-- [x] 表示选中 --> - [x] 新特性 - [ ] Bug修复 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [x] AI辅助编写 See merge request: cann/ops-nn!11335
22 天前
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
2 个月前
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
2 个月前
CMakeLists.txt
DynamicQuant等22个算子支持Ascend960dt Co-authored-by: TH_HW<taohai2@huawei.com> # message auto-generated for no-merge-commit merge: !11335 merge backBlue into master DynamicQuant等22个算子支持Ascend960dt Created-by: TH_HW Commit-by: TH_HW Merged-by: cann-robot Description: ## 描述 （+17812 / -2151）为一批 ops-nn 算子补齐 **Ascend 960DT（NPU Arch 9201）** 平台支持，并重构 GeGluV2 arch35 内核、为 Erfinv 增加 inplace 通路。 主要改动如下： 1. **Ascend960DT 平台适配（22 个算子）** - activation：Relu、ReluV2、ReluGrad、ReluGradV2、Sigmoid、SigmoidGrad、PRelu、PReluGradUpdate、SiluGrad、Softplus、Swish、SwishGrad、SwiGluGrad、GeGluV2。 - index：MapIndex、SparseSlice、QuantUpdateScatter。 - quant：Quantize、DynamicQuant、DynamicQuantV2、AntiMxQuant、GroupedDynamicMxQuant。 - 具体做法：CMakeLists.txt 的 SUPPORT_COMPUTE_UNIT / SUPPORT_TILING_DIR 增加 ascend960dt；*_def.cpp 增加 AddConfig("ascend960dt", ...)；aclnn/op_api 与 infershape 增加 NpuArch::DAV_9201 / Ascend960DT 分支；kernel 增加 __NPU_ARCH__ == 9201 分支；新增 op_host/config/ascend960dt/ 下的算子二进制配置（共 22 个，如 relu_binary.json、p_relu.json 等）；scripts/kernel/binary_config/ascendc_config.json 为相关算子登记 ascend960dt 计算单元与编译选项。 2. **GeGluV2 arch35 内核重构** - 将 arch35 模板按 erf / tanh 两类拆分，并补齐 align、align_last_axis_big、vreduce × fp16/bf16/fp32 变体，统一命名 *_arch35.h。 - tiling 新增 RegBase per-template per-element UB 预算模型（GetRegbasePerElem），并支持 ascend960dt。 - aclnnGeGlu 参数校验重构（CheckShape 支持 0D 输入与 dim 范围校验），新增 ST 用例（all_aclnnGeGlu(.V3).json）并调整 UT。 3. **Erfinv 支持 inplace**：新增 inplace dtype 支持列表与校验（GetInplaceDtypeSupportList / CheckInplaceDtypeValid），并补充 op_api UT。 4. **内核与接口修复** - SparseSlice：9201 路径改用 SetValue 写入 sliceStart/sliceEnd（替代 DataCopy），Reg::Reduce 使用完整命名空间限定。 - QuantUpdateScatter：9201 增加 FLOAT_OVERFLOW_MODE_CTRL 保护。 - ReluV2 infershape：Ascend960DT 不支持 NC1HWC0，UINT1 支持列表加入 Ascend960DT。 - DynamicQuant / Mx 量化：9201 上 fp8 scale 改为 bf16 乘法后再 cast 到 e8m0。 5. **代码规范**：统一 License 头注释格式、补齐文件末尾换行、调整 quant 系列 UT 的 CMakeLists；新增 MapIndex、SparseSlice 的 GE IR 图模式示例。 ## 关联的Issue 无。 ## 测试 - 新增 GeGluV2 ST 用例与 Erfinv inplace UT，补充/调整 GeGluV2 相关 UT。 - 新增 MapIndex / SparseSlice 的 GE IR 示例（examples/test_geir_*.cpp）。 - 通过 ascend960dt 编译与静态检查。 ## 文档更新 无。 ## 类型标签 <!-- [x] 表示选中 --> - [x] 新特性 - [ ] Bug修复 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [x] AI辅助编写 See merge request: cann/ops-nn!11335
22 天前
README.md
修正 Swish README 中 Atlas A2 的产品支持标记 Co-authored-by: KaranocaVe<karanocave@qq.com> # message auto-generated for no-merge-commit merge: !12212 merge fix/2026-10-15-ops-nn-swish-atlas-a2-platform-doc-01 into master 修正 Swish README 中 Atlas A2 的产品支持标记 Created-by: KaranocaVe Commit-by: KaranocaVe Merged-by: cann-robot Description: # 修正 Swish README 中 Atlas A2 的产品支持标记 ## 变更 activation/swish/README.md 将 Atlas A2 系列产品的支持标记由 √ 改为 ×。Swish Host OpDef 仅注册 ascend950 和 ascend350，构建目标也未包含对应的 ascend910b，因此 README 不应将 Atlas A2 标记为支持。 ## 验证 - 验证结果：Swish Host OpDef 仅注册 ascend950、ascend350，CMake 构建目标不含 ascend910b，而 Atlas A2 的 README 行标为支持。 - 验证结果：补丁只把 README 的 Atlas A2 标记由 √ 改为 ×，没有改动算子实现。 - 未运行构建或设备用例；结论限于文档与注册/构建目标的一致性。 关联 Issue: https://gitcode.com/cann/ops-nn/issues/6844 See merge request: cann/ops-nn!12212
21 天前
README

Swish

产品支持情况

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

功能说明

  • 算子功能：计算Swish激活函数值。

  • 计算公式：

    y=x1+e−scale×xy = \frac{x}{1 + e^{-\text{scale} \times x}}

    其中：

    • xx表示输入张量
    • scale\text{scale}表示缩放参数，默认值为1.0
    • ee表示自然对数的底数
    • yy表示输出张量

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
x 输入 表示输入张量，支持1D-8D维度。 FLOAT16、FLOAT、BFLOAT16 ND
y 输出 表示输出张量，与输入x具有相同的类型、形状和格式。 与x一致 ND
scale 属性 表示缩放参数，用于调整Swish函数的形状。默认值为1.0。 FLOAT -

约束说明

无

调用说明

调用方式 调用样例 说明
aclnn调用 test_aclnn_swish 通过aclnnSwish接口方式调用Swish算子。