本项目是CANN提供的神经网络类计算算子库,实现网络在NPU上加速计算。
| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
子流水线支持更新门禁代码 Co-authored-by: jinyijian<jinyijian1@huawei.com> # message auto-generated for no-merge-commit merge: !4515 merge CI into master 子流水线支持更新门禁代码 Created-by: jinyijian Commit-by: jinyijian Merged-by: cann-robot Description: ## 描述 子流水线支持更新门禁代码 当门禁包含子流水线时,如果没有这个配置,会导致子流水线中任务无法更新最新代码,使用落后的代码跑门禁 ## 关联的Issue ## 测试 ## 文档更新 无 ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!4515 | 5 个月前 | |
fix ut problem Co-authored-by: ni-zhihao<nizhihao@h-partners.com> # message auto-generated for no-merge-commit merge: !4489 merge master into master fix ut problem Created-by: ni-zhihao Commit-by: ni-zhihao Merged-by: cann-robot Description: ## 描述 fix ut problem ## 关联的Issue 关联Issue [#2232](https://gitcode.com/cann/ops-nn/issues/2232) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!4489 | 5 个月前 | |
减少add_modules_sources的必要参数,降低函数复杂度 Co-authored-by: yang-di52<yangdi52@huawei.com> # message auto-generated for no-merge-commit merge: !4456 merge issue_fix into master 减少add_modules_sources的必要参数,降低函数复杂度 Created-by: yang-di52 Commit-by: yang-di52 Merged-by: cann-robot Description: ## 描述 * 变更概要 优化 add_modules_sources CMake 函数,将 DIR 参数从必选改为可选,降低函数调用复杂度。关联 Issue #2071。 * 主要改动 cmake/func.cmake: 将 DIR 参数从 oneValueArgs 移至 multiValueArgs,并添加默认值逻辑,未指定时使用当前源码目录 ## 关联的Issue [#2071](https://gitcode.com/cann/ops-nn/issues/2071) ## 测试 已通过修改scripts/ci/mirror_update_time.txt 执行全量门禁通过 ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!4456 | 5 个月前 | |
新增Gather和TensorScatterAdd算子原型 Co-authored-by: 张鑫<zhangxin660@h-partners.com> # message auto-generated for no-merge-commit merge: !4461 merge add_REG_OP into master 新增Gather和TensorScatterAdd算子原型 Created-by: z30075199 Commit-by: 张鑫 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动,包括改动的原因和所采取的方法。--> 新增Gather和TensorScatterAdd算子原型 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。例如:关联Issue #000--> https://gitcode.com/cann/ops-nn/issues/2445 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!4461 | 5 个月前 | |
geir用例修改 Co-authored-by: Ding_Jing<dingjing19@huawei.com> # message auto-generated for no-merge-commit merge: !3795 merge geir_modify_0414 into master geir用例修改 Created-by: Ding_Jing Commit-by: Ding_Jing Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动,包括改动的原因和所采取的方法。--> 本PR针对geir用例修改: 1、删除无用头文件 2、用例报错时再打Error Message信息,防止产生误导 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。例如:关联Issue #000--> issue [#2045](https://gitcode.com/cann/ops-nn/issues/2045) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> Assert: [2026-04-14 09:48:24] 2026-04-14 09:48:24,000 - INFO - [XIR]: Start to add compute graph to ir session [2026-04-14 09:48:24] 2026-04-14 09:48:24,000 - INFO - [XIR]: Session add ir compute graph to ir session success [2026-04-14 09:48:24] 2026-04-14 09:48:24,000 - INFO - [XIR]: dump graph to txt [2026-04-14 09:48:24] 2026-04-14 09:48:24,000 - INFO - [XIR]: Start to run ir compute graph [2026-04-14 09:48:24] 2026-04-14 09:48:24,000 - INFO - [XIR]: Session run ir compute graph success [2026-04-14 09:48:25] 2026-04-14 09:48:24,000 - INFO - [XIR]: Start to finalize ir graph session [2026-04-14 09:48:27] 2026-04-14 09:48:25,000 - INFO - [XIR]: Finalize ir graph session success [2026-04-14 09:48:27] [2026-04-14 09:48:27] ---------------------------------------------------------------- [2026-04-14 09:48:27] Run test_geir_assert success. [2026-04-14 09:48:27] ---------------------------------------------------------------- [2026-04-14 09:48:27] Identity: [2026-04-14 09:49:31] 2026-04-14 09:49:31,000 - INFO - [XIR]: Start to add compute graph to ir session [2026-04-14 09:49:31] 2026-04-14 09:49:31,000 - INFO - [XIR]: Session add ir compute graph to ir session success [2026-04-14 09:49:31] 2026-04-14 09:49:31,000 - INFO - [XIR]: dump graph to txt [2026-04-14 09:49:31] 2026-04-14 09:49:31,000 - INFO - [XIR]: Start to run ir compute graph [2026-04-14 09:49:31] 2026-04-14 09:49:31,000 - INFO - [XIR]: Session run ir compute graph success [2026-04-14 09:49:32] result[0] is: 2.000000 [2026-04-14 09:49:32] result[1] is: 2.000000 [2026-04-14 09:49:32] result[2] is: 2.000000 [2026-04-14 09:49:32] result[3] is: 2.000000 [2026-04-14 09:49:32] result[4] is: 2.000000 [2026-04-14 09:49:32] result[5] is: 2.000000 [2026-04-14 09:49:32] result[6] is: 2.000000 [2026-04-14 09:49:32] result[7] is: 2.000000 [2026-04-14 09:49:32] 2026-04-14 09:49:31,000 - INFO - [XIR]: Start to finalize ir graph session [2026-04-14 09:49:34] 2026-04-14 09:49:33,000 - INFO - [XIR]: Finalize ir graph session success [2026-04-14 09:49:34] [2026-04-14 09:49:34] ---------------------------------------------------------------- [2026-04-14 09:49:34] Run test_geir_identity success. [2026-04-14 09:49:34] ---------------------------------------------------------------- [2026-04-14 09:49:34] GatherV2: [2026-04-14 09:51:31] result[0] is: (1.500000, 11.500000) [2026-04-14 09:51:31] result[1] is: (3.500000, 13.500000) [2026-04-14 09:51:31] result[2] is: (6.500000, 16.500000) [2026-04-14 09:51:31] result[3] is: (8.500000, 18.500000) [2026-04-14 09:51:31] result[4] is: (11.500000, 21.500000) [2026-04-14 09:51:31] result[5] is: (13.500000, 23.500000) [2026-04-14 09:51:33] [2026-04-14 09:51:33] ---------------------------------------------------------------- [2026-04-14 09:51:33] Run test_geir_gather_v2 success. [2026-04-14 09:51:33] ---------------------------------------------------------------- [2026-04-14 09:51:33] reverse_sequence: [2026-04-14 10:30:09] === ReverseSequence GEIR Test Start === [2026-04-14 10:30:09] output 0 dtype : 11 [2026-04-14 10:30:10] INFO - [XIR]: Precison is ok. [2026-04-14 10:30:13] === Test completed successfully === [2026-04-14 10:30:13] [2026-04-14 10:30:13] ---------------------------------------------------------------- [2026-04-14 10:30:13] Run test_geir_reverse_sequence success. [2026-04-14 10:30:13] ---------------------------------------------------------------- [2026-04-14 10:30:13] ScatterElements: [2026-04-14 09:56:49] result[0] is: (1.000000, 1.000000) [2026-04-14 09:56:49] result[1] is: (10.000000, -1.000000) [2026-04-14 09:56:49] result[2] is: (3.000000, 3.000000) [2026-04-14 09:56:49] result[3] is: (20.000000, -2.000000) [2026-04-14 09:56:49] result[4] is: (5.000000, 5.000000) [2026-04-14 09:56:51] [2026-04-14 09:56:51] ---------------------------------------------------------------- [2026-04-14 09:56:51] Run test_geir_scatter_elements success. [2026-04-14 09:56:51] ---------------------------------------------------------------- [2026-04-14 09:56:51] TensorScatterUpdate: [2026-04-14 09:57:50] 2026-04-14 09:57:50,000 - INFO - [XIR]: Start to add compute graph to ir session [2026-04-14 09:57:50] 2026-04-14 09:57:50,000 - INFO - [XIR]: Session add ir compute graph to ir session success [2026-04-14 09:57:50] 2026-04-14 09:57:50,000 - INFO - [XIR]: dump graph to txt [2026-04-14 09:57:50] 2026-04-14 09:57:50,000 - INFO - [XIR]: Start to run ir compute graph [2026-04-14 09:57:50] 2026-04-14 09:57:50,000 - INFO - [XIR]: Session run ir compute graph success [2026-04-14 09:57:50] result[0] is: (100.000000, 10.000000) [2026-04-14 09:57:50] result[1] is: (101.000000, 11.000000) [2026-04-14 09:57:50] result[2] is: (102.000000, 12.000000) [2026-04-14 09:57:50] result[3] is: (103.000000, 13.000000) [2026-04-14 09:57:50] result[4] is: (104.000000, 14.000000) [2026-04-14 09:57:50] result[5] is: (5.000000, -2.500000) [2026-04-14 09:57:50] result[6] is: (6.000000, 3.500000) [2026-04-14 09:57:50] result[7] is: (7.000000, -3.500000) [2026-04-14 09:57:50] result[8] is: (8.000000, 4.500000) [2026-04-14 09:57:50] result[9] is: (9.000000, -4.500000) [2026-04-14 09:57:50] result[10] is: (200.000000, 20.000000) [2026-04-14 09:57:50] result[11] is: (201.000000, 21.000000) [2026-04-14 09:57:50] result[12] is: (202.000000, 22.000000) [2026-04-14 09:57:50] result[13] is: (203.000000, 23.000000) [2026-04-14 09:57:50] result[14] is: (204.000000, 24.000000) [2026-04-14 09:57:50] 2026-04-14 09:57:50,000 - INFO - [XIR]: Start to finalize ir graph session [2026-04-14 09:57:53] 2026-04-14 09:57:52,000 - INFO - [XIR]: Finalize ir graph session success [2026-04-14 09:57:53] [2026-04-14 09:57:53] ---------------------------------------------------------------- [2026-04-14 09:57:53] Run test_geir_tensor_scatter_update success. [2026-04-14 09:57:53] ---------------------------------------------------------------- Where: [2026-04-14 10:03:13] 2026-04-14 10:03:13,000 - INFO - [XIR]: Start to add compute graph to ir session [2026-04-14 10:03:13] 2026-04-14 10:03:13,000 - INFO - [XIR]: Session add ir compute graph to ir session success [2026-04-14 10:03:13] 2026-04-14 10:03:13,000 - INFO - [XIR]: dump graph to txt [2026-04-14 10:03:13] 2026-04-14 10:03:13,000 - INFO - [XIR]: Start to run ir compute graph [2026-04-14 10:03:13] 2026-04-14 10:03:13,000 - INFO - [XIR]: Session run ir compute graph success [2026-04-14 10:03:14] 2026-04-14 10:03:13,000 - INFO - [XIR]: Precision is ok [2026-04-14 10:03:14] 2026-04-14 10:03:13,000 - INFO - [XIR]: Start to finalize ir graph session [2026-04-14 10:03:16] 2026-04-14 10:03:15,000 - INFO - [XIR]: Finalize ir graph session success [2026-04-14 10:03:16] [2026-04-14 10:03:16] ---------------------------------------------------------------- [2026-04-14 10:03:16] Run test_geir_where success. [2026-04-14 10:03:16] ---------------------------------------------------------------- [2026-04-14 10:03:16] ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> 针对geir用例修改,不涉及文档更新 ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!3795 | 5 个月前 | |
卷积反向算子适配load3d基础api接口名变更,toolkit version隔离 Co-authored-by: zhaozhongyao<zhaozhongyao2@huawei.com> # message auto-generated for no-merge-commit merge: !4523 merge load3d0506 into master 卷积反向算子适配load3d基础api接口名变更,toolkit version隔离 Created-by: zhaozhongyao Commit-by: zhaozhongyao Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动,包括改动的原因和所采取的方法。--> 卷积反向算子适配load3d基础api接口名变更,toolkit version隔离 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。例如:关联Issue #000--> https://gitcode.com/cann/ops-nn/issues/2517 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!4523 | 5 个月前 | |
modified md files(for readability improvement) Co-authored-by: duhuiping<duhuiping@h-partners.com> # message auto-generated for no-merge-commit merge: !4511 merge master_md into master modified md files(for readability improvement) Created-by: gitee-duhuiping Commit-by: duhuiping Merged-by: cann-robot Description: ## 描述 modified md files(for readability improvement) ## 关联的Issue Data issue ## 测试 Optimizing the data format ## 文档更新 md files ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!4511 | 5 个月前 | |
补充add_example的infershape ut Co-authored-by: yang-di52<yangdi52@huawei.com> # message auto-generated for no-merge-commit merge: !4533 merge issue_fix2 into master 补充add_example的infershape ut Created-by: yang-di52 Commit-by: yang-di52 Merged-by: cann-robot Description: ## 描述 ### 背景 当前代码仓内提供的add_example样例缺少infershape ut ### 主要修改内容 增加test_add_example_infershape.cpp,为add_example补充缺少infershape ut ## 关联的Issue [#2471](https://gitcode.com/cann/ops-nn/issues/2471) ## 测试  ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!4533 | 5 个月前 | |
SoftplusV2Grad算子Ascend C实现 Co-authored-by: fulltower<full.fu@huawei.com> # message auto-generated for no-merge-commit merge: !4014 merge softplusv2grad into master SoftplusV2Grad算子Ascend C实现 Created-by: fullt Commit-by: fulltower Merged-by: cann-robot Description: ## 描述 参考版本内置Sigmoid算子的TBE实现,在昇腾NPU上使用Ascend C编程语言实现相同功能的算子。 ## 关联的Issue https://gitcode.com/cann/ops-nn/issues/229 ## 测试 aclnn测试 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!4014 | 5 个月前 | |
modified md files(for readability improvement) Co-authored-by: duhuiping<duhuiping@h-partners.com> # message auto-generated for no-merge-commit merge: !4511 merge master_md into master modified md files(for readability improvement) Created-by: gitee-duhuiping Commit-by: duhuiping Merged-by: cann-robot Description: ## 描述 modified md files(for readability improvement) ## 关联的Issue Data issue ## 测试 Optimizing the data format ## 文档更新 md files ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!4511 | 5 个月前 | |
[kernel] embedding_hash_table and sparse_tensor_dense_mat_mul modify simt api | 5 个月前 | |
add bucketizev2 geir example Co-authored-by: lixin433<lixin433@huawei.com> # message auto-generated for no-merge-commit merge: !4308 merge bucketize into master add bucketizev2 geir example Created-by: lixin433 Commit-by: lixin433 Merged-by: cann-robot Description: ## 描述 增加bucketize_v2 geir examples,修正部分文档 <!--在这里详细描述你的改动,包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。例如:关联Issue #000--> 关联Issue [#2494](https://gitcode.com/cann/ops-nn/issues/2494) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> examples已经测试,冒烟已经通过 ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!4308 | 5 个月前 | |
修复logit、logit_grad示例注释问题 Co-authored-by: ni-zhihao<nizhihao@h-partners.com> # message auto-generated for no-merge-commit merge: !4438 merge master into master 修复logit、logit_grad示例注释问题 Created-by: ni-zhihao Commit-by: ni-zhihao Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动,包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。例如:关联Issue #000--> ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!4438 | 5 个月前 | |
[kernel] embedding_hash_table and sparse_tensor_dense_mat_mul modify simt api | 5 个月前 | |
补充 Ascend 950 AddRmsNorm 算子的完整参数组合说明 Co-authored-by: liangyuhua<liangyuhua4@huawei.com> # message auto-generated for no-merge-commit merge: !4559 merge 0507 into master 补充 Ascend 950 AddRmsNorm 算子的完整参数组合说明 Created-by: liangyuhua Commit-by: liangyuhua Merged-by: cann-robot Description: ## 描述 补充 Ascend 950 AddRmsNorm 算子的完整参数组合说明。 **问题**: - 文档中缺少 Ascend 950 的完整参数组合表 - 原文档只说明了标准模式的 gamma shape 要求,未说明 PRE_RMS_NORM 和 POST_RMS_NORM 模式的特殊要求 **修改**: - 删除第176-178行不完整的单独说明 - 在第265行将 Ascend 950 追加到 A2/A3 产品列表,共用同一个组合表 **影响**: - 用户可以准确找到 950 支持的所有参数组合 - 避免因文档不完整导致的参数配置错误 ## 关联的Issue #2536 ## 测试 文档更新,已通过本地验证: - 文档格式正确,表格显示正常 - 内容与代码实现一致(通过代码分析验证) - 修改逻辑合理,符合文档规范 ## 文档更新 已更新 norm/add_rms_norm/docs/aclnnAddRmsNorm.md ## 类型标签 - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!4559 | 5 个月前 | |
修改LOGEreason内容 Co-authored-by: tianqiguang<tianqiguang@huawei.com> # message auto-generated for no-merge-commit merge: !4391 merge master into master 修改LOGEreason内容 Created-by: tianqiguang Commit-by: tianqiguang Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动,包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。例如:关联Issue #000--> ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!4391 | 5 个月前 | |
清除AdaptiveAvgPool2d代码中遗留注释 Co-authored-by: SGQT<shangguanqinnan@huawei.com> # message auto-generated for no-merge-commit merge: !4459 merge fix_adaavgpool2d_comment into master 清除AdaptiveAvgPool2d代码中遗留注释 Created-by: u010470851 Commit-by: SGQT Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动,包括改动的原因和所采取的方法。--> 文件:pooling/adaptive_avg_pool2d/op_kernel/arch35/adaptive_pool2d_big_kernel.h,第 99 行有误提交的遗留TODO注释 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。例如:关联Issue #000--> 关联的issue https://gitcode.com/cann/ops-nn/issues/2086 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> 删除注释无风险 ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他,请描述: 清除遗留注释 See merge request: cann/ops-nn!4459 | 5 个月前 | |
dynamic_quant_v2算子新增pertensor量化模式的groupIndex支持 Co-authored-by: LiWeiJian16<liweijian16@huawei.com> # message auto-generated for no-merge-commit merge: !3976 merge dynamic_quant_moe_pertensor into master dynamic_quant_v2算子新增pertensor量化模式的groupIndex支持 Created-by: liweijian16 Commit-by: liweijian16;LiWeiJian16 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动,包括改动的原因和所采取的方法。--> 本次PR扩展了动态量化算子对MoE(Mixture of Experts)模型场景的支持,核心解决了算子原有pertensor(按张量)量化模式无法处理分组索引(group_index)的问题。通过本次修改,动态量化算子能够正确支持MoE架构下按张量进行全局缩放因子计算的量化需求,从而满足更复杂模型的量化任务。 修改方案描述 修改方案分为框架层和内核实现层。在框架层,移除了输入校验中对pertensor模式不支持group_index的限制,并新增了TPL_MOE_PER_TENSOR_FULL_LOAD和TPL_MOE_PER_TENSOR_LARGE_SHAPE两种量化模式枚举,扩展了内核分发逻辑以支持新模式。在内核实现层,新增了两个针对MoE场景优化的高性能量化内核:DynamicQuantRegbaseMoeFullLoadPertensor和DynamicQuantMoeLargeShapePertensor。它们均采用向量化指令优化,支持分组平滑,通过分阶段计算全局缩放因子和偏移量,并最终完成数据量化输出,以高效处理不同数据规模的MoE模型。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。例如:关联Issue #000--> https://gitcode.com/cann/ops-nn/issues/2281 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> - [x] 泛化用例 (500条) - [x] 网络用例 (92条) - [x] 二级冒烟 ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> 不涉及文档更新。 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!3976 | 5 个月前 | |
doc Tools扫描资料低错问题修改 Co-authored-by: gitee-yanglulu<yanglulul@h-partners.com> # message auto-generated for no-merge-commit merge: !4231 merge master into master doc Tools扫描资料低错问题修改 Created-by: gitee-yanglulu Commit-by: gitee-yanglulu Merged-by: cann-robot Description: doc Tools扫描资料低错问题修改 See merge request: cann/ops-nn!4231 | 5 个月前 | |
add aicpu const folding ops flow. Co-authored-by: liu-wei<lovline.liuwei@huawei.com> # message auto-generated for no-merge-commit merge: !4087 merge master into master add aicpu const folding ops flow. Created-by: liu-wei Commit-by: liu-wei Merged-by: cann-robot Description: ## 描述 新增 nn const folding ops flow(常量折叠算子流程),用于支持图像处理算子的常量折叠优化,减少运行时计算开销,提升推理性能。 ## 关联的Issue https://gitcode.com/cann/ops-nn/issues/2532 ## 测试 根据代码变更,测试场景如下: 1. **编译构建测试** - 执行 cmake 配置和编译,验证 CMake 配置变更正确 - 验证 cmake/func.cmake、symbol.cmake、variables.cmake 变更正确加载 2. **算子加载测试** - 验证 crop_and_resize 算子能正确加载 - 验证 OpsCvInc.xml 中新增算子接口注册成功 3. **功能验证** - 验证常量折叠功能正常工作 - 确认推理时常量折叠优化生效 ## 文档更新 无。 ## 类型标签 - [ ] 🐛 Bug 修复 - [x] ✨ 新特性 - [x] ⚡ 性能优化 - [ ] ♻️ 重构 - [ ] 🧪 测试 - [x] 📦 构建/CI - [x] 🔧 配置变更 - [ ] 📝 文档更新 - [ ] ⬆️ 依赖升级 - [ ] 🔒 安全修复 - [ ] 🧹 代码清理 - [ ] ❓ 其他,请描述: See merge request: cann/ops-nn!4087 | 5 个月前 | |
qbmmv4 api quant_batch_matmul_v3/v4 ut重构 Co-authored-by: zhoushaolong<zhoushaolong3@huawei.com> # message auto-generated for no-merge-commit merge: !3900 merge ut/0413 into master qbmmv4 api quant_batch_matmul_v3/v4 ut重构 Created-by: zhoushaolong Commit-by: zhoushaolong Merged-by: cann-robot Description: ## 描述 将 quant_batch_matmul_v3 和 quant_batch_matmul_v4 相关 op_host / op_api UT 统一改为 CSV 驱动模式,补充并整理对应 CSV 用例。 新增公共字符串解析能力 tests/ut/common/ut_string_utils.h,统一封装 Trim、字符串切分、bool 解析、dims 解析、默认 stride 构造等 UT 常用基础能力,减少各个测试文件内重复解析代码。 调整 quant_batch_matmul_v3 的 ACLNN op_api、infershape、tiling UT,以及 quant_batch_matmul_v4 的 tiling、infershape、pergroup、pertile、A8W4、QuantMatmulV5 op_api UT 的用例加载方式,统一通过 GetParams / LoadParamsFromCsv 从 CSV 读取参数并实例化参数化测试。 补充 test_aclnn_quant_matmul_v3.csv、test_aclnn_quant_matmul_v4.csv、test_aclnn_quant_matmul_weight_nz.csv、test_quant_batch_matmul_v3_infershape.csv 以及多类 quant_batch_matmul_v4 CSV 用例,清理旧的重复手写用例逻辑,提升 UT 可读性和维护性。 同步修复流水线中 CSV 路径解析、参数化用例空实例化、V4 caseName 解析和 V3 Ascend950 多线程 UT 稳定性问题。 本次修改范围仅限 UT、CSV 用例和测试公共代码,不修改算子实现逻辑。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。例如:关联Issue #000--> https://gitcode.com/cann/ops-nn/issues/2150 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> 本地跑了下面命令: bash build.sh -u --ophost --ops=quant_batch_matmul_v4 --cov bash build.sh -u --opapi --ops=quant_batch_matmul_v3 --cov bash build.sh -u --ophost --ops=quant_batch_matmul_v3 --cov 用例都PASS了 ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他,请描述:ut重构 See merge request: cann/ops-nn!3900 | 5 个月前 | |
修改整仓产品名问题及标签问题 Co-authored-by: gitee-yanglulu<yanglulul@h-partners.com> # message auto-generated for no-merge-commit merge: !4204 merge master into master 修改整仓产品名问题及标签问题 Created-by: gitee-yanglulu Commit-by: gitee-yanglulu Merged-by: cann-robot Description: 修改整仓产品名问题及标签问题 See merge request: cann/ops-nn!4204 | 5 个月前 | |
修改clang format Co-authored-by: chenqi317<chenqi317@huawei.com> # message auto-generated for no-merge-commit merge: !4385 merge master into master 修改clang format Created-by: chenqi317 Commit-by: chenqi317 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动,包括改动的原因和所采取的方法。--> 修改clang-format 规划 1、枚举的{ 跟enum 同行 2、string开头不强制换行 3、短函数实现允许同行 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。例如:关联Issue #000--> ## 测试 <!--描述进行了哪些测试来验证你的改动。--> 不涉及 ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> 不涉及 ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!4385 | 5 个月前 | |
fix foreach tiling shape check Co-authored-by: alfengyuan<yuanyaofeng@h-partners.com> # message auto-generated for no-merge-commit merge: !3200 merge master into master fix foreach tiling shape check Created-by: alfengyuan Commit-by: alfengyuan Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动,包括改动的原因和所采取的方法。--> 本次PR主要包含两项优化和一项缺陷修复。首先,在代码库配置层面,为保持仓库整洁,添加了对ModelArts平台临时目录的忽略规则。其次,在代码逻辑层面,移除了一个算子实现中冗余的函数调用以简化流程。核心修复是针对Foreach算子输出形状检查逻辑中的一个缺陷,原逻辑过于严格,错误地限制了输出形状必须与输入完全一致,本次修复放宽了条件,允许输出形状与输入不同但元素总数不大于输入,从而支持了更灵活的操作场景。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。例如:关联Issue #000--> https://gitcode.com/cann/ops-nn/issues/1738 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!3200 | 6 个月前 | |
增加pre-commit配置 Co-authored-by: jinyijian<jinyijian1@huawei.com> # message auto-generated for no-merge-commit merge: !3469 merge CI into master 增加pre-commit配置 Created-by: jinyijian Commit-by: jinyijian Merged-by: cann-robot Description: ## 描述 新增了pre-commit的yaml配置 ## 关联的Issue [1908](https://gitcode.com/cann/ops-nn/issues/1908) ## 测试 本地构造不同文件,测试拦截场景 ## 文档更新 docs/zh/develop/pre-commit_guide.md ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!3469 | 6 个月前 | |
修改doc tools低错 Co-authored-by: caiwenwen<caiwenwen6@h-partners.com> # message auto-generated for no-merge-commit merge: !4214 merge master into master 修改doc tools低错 Created-by: caiwenwen Commit-by: caiwenwen Merged-by: cann-robot Description: ## 描述 修改nn仓的markdown语法、htlm是否闭合、链接是否可以正常跳转、本地链接是否正常跳转修改低错修改 ## 关联的Issue 关联Issue [#1783](https://gitcode.com/cann/ops-nn/issues/1783) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 全部文档的链接跳转、htlm标签闭合、markdown语法规范问题 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [X] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!4214 | 5 个月前 | |
refactor: 引用工程公共仓cann-cmake统一打包构建API,移除本地冗余脚本 Co-authored-by: yang-di52<yangdi52@huawei.com> # message auto-generated for no-merge-commit merge: !4109 merge bugfix1 into master refactor: 引用工程公共仓cann-cmake统一打包构建API,移除本地冗余脚本 Created-by: yang-di52 Commit-by: yang-di52 Merged-by: cann-robot Description: ## 描述 重构 ops-nn 仓库的打包构建流程,将本地维护的打包脚本和安装脚本替换为引用工程公共仓 cann-cmake 中统一的 API,降低构建工程的复杂度和维护成本。 ### 主要改动: 1. 升级 cann-cmake 版本:从 master-001 升级到 master-002,新增 SHA256 校验(fetch_cann_cmake.cmake) 2. 替换打包脚本引用路径:将 cmake/package.cmake 中所有 ${CMAKE_SOURCE_DIR}/scripts/package/ 和 ${CMAKE_SOURCE_DIR}/cmake/ 本地路径替换为 ${CANN_CMAKE_DIR}/ 公共路径 3. 移除本地冗余脚本:删除 cmake/makeself_built_in.cmake、cmake/makeself_custom.cmake、cmake/third_party/makeself-fetch.cmake 以及 scripts/package/common/ 和 scripts/package/latest_manager/ 下的全部 21 个文件 3. 更新打包配置:新增 CPACK_PACKAGE_PARAM_NAME 参数和 share_info_name XML 配置项(ops_math.xml) 4. 更新编译文档:同步更新 cann-cmake 版本号说明(docs/zh/install/compile.md) ## 关联的Issue [#2381](https://gitcode.com/cann/ops-nn/issues/2381) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 docs/zh/install/compile.md ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他,请描述:构建重构 See merge request: cann/ops-nn!4109 | 5 个月前 | |
修改doc tools低错 Co-authored-by: caiwenwen<caiwenwen6@h-partners.com> # message auto-generated for no-merge-commit merge: !4214 merge master into master 修改doc tools低错 Created-by: caiwenwen Commit-by: caiwenwen Merged-by: cann-robot Description: ## 描述 修改nn仓的markdown语法、htlm是否闭合、链接是否可以正常跳转、本地链接是否正常跳转修改低错修改 ## 关联的Issue 关联Issue [#1783](https://gitcode.com/cann/ops-nn/issues/1783) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 全部文档的链接跳转、htlm标签闭合、markdown语法规范问题 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [X] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!4214 | 5 个月前 | |
init | 1 年前 | |
规范化OAT.xml中许可证声明形式 Co-authored-by: chenyx2012<975102023@qq.com> # message auto-generated for no-merge-commit merge: !3385 merge master into master 规范化OAT.xml中许可证声明形式 Created-by: chenyx_2012 Commit-by: chenyx2012 Merged-by: cann-robot Description: ## 描述 规范化OAT.xml中许可证声明形式 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。例如:关联Issue #000--> ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!3385 | 6 个月前 | |
nn仓README.md里的index_fill跳转链接修复&添加index_fill算子文档 Co-authored-by: klein8793<zhangli24@huawei.com> # message auto-generated for no-merge-commit merge: !4509 merge dev_zl_2026_0506 into master nn仓README.md里的index_fill跳转链接修复&添加index_fill算子文档 Created-by: klein8793 Commit-by: klein8793 Merged-by: cann-robot Description: ## 描述 nn仓README.md里的index_fill跳转链接修复&添加index_fill算子文档 ## 关联的Issue 关联Issue [#2486](https://gitcode.com/cann/ops-nn/issues/2486) ## 测试 NA ## 文档更新 README.md index/index_fill/README.md ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!4509 | 5 个月前 | |
nn仓doc tools扫描内容修改 Co-authored-by: caiwenwen<caiwenwen6@h-partners.com> # message auto-generated for no-merge-commit merge: !3414 merge master into master nn仓doc tools扫描内容修改 Created-by: caiwenwen Commit-by: caiwenwen Merged-by: cann-robot Description: ## 描述 修改markdown语法和链接,1000个左右。 ## 关联的Issue 关联Issue [#1783](https://gitcode.com/cann/ops-nn/issues/1783) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 更新了README.md、gitcode、cmake、common、control、conv、docs、examples、experimental、hash、index、loss、matmul、optim、pooling等文件。 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!3414 | 6 个月前 | |
fix Third_Party_Open_Source_Software_List.yaml Co-authored-by: liuyufan0725<liuyufan9@huawei.com> # message auto-generated for no-merge-commit merge: !1879 merge fix_yaml into master fix Third_Party_Open_Source_Software_List.yaml Created-by: liuyufan0725 Commit-by: liuyufan0725 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动,包括改动的原因和所采取的方法。--> 更新三方依赖清单Third_Party_Open_Source_Software_List.yaml 1、增加protobuf 2、将nlohmann/json修改为json ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。例如:关联Issue #000--> 关联Issue [#1042](https://gitcode.com/cann/ops-nn/issues/1042) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> 不涉及 ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> 更新了Third_Party_Open_Source_Software_List.yaml ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!1879 | 7 个月前 | |
【描 述】 eigen update to 5.0.0 Co-authored-by: j00427146<jialimin1@huawei.com> # message auto-generated for no-merge-commit merge: !1152 merge master into master 【描 述】 eigen update to 5.0.0 Created-by: jialimin1 Commit-by: j00427146 Merged-by: cann-robot Description: ## 描述 现开源软件eigen 3.4.0 EOM,Eigen 从3.4.0版本升级到5.0.0,解决生命周期问题。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。例如:关联Issue #000--> <!-- 如果这个PR是为了解决特定的问题单,请在这里描述问题单单号。--> https://gitcode.com/cann/ops-nn/issues/975 https://gitcode.com/cann/ops-nn/issues/389 ## 测试   ## 文档更新 SECURITY.md ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他,请描述:Update Eigen to 5.0.0 See merge request: cann/ops-nn!1152 | 8 个月前 | |
增加mssanitizer参数 Co-authored-by: yang-di52<yangdi52@huawei.com> # message auto-generated for no-merge-commit merge: !4531 merge issue_fix3 into master 增加mssanitizer参数 Created-by: yang-di52 Commit-by: yang-di52 Merged-by: cann-robot Description: ## 描述 ### 背景 当前nn仓build脚本没有支持--mssanitizer参数 ### 主要修改 * 增加--mssanitizer支持 * 在build脚本的help资料中增加--mssanitizer描述 ## 关联的Issue [#2515](https://gitcode.com/cann/ops-nn/issues/2515) ## 测试 *注* :使用--mssanitizer参数需要安装完整的toolkit的包   ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!4531 | 5 个月前 | |
feat: 预修改控制类算子迁移 Co-authored-by: 叶龙剑<yelongjian1@huawei.com> # message auto-generated for no-merge-commit merge: !4363 merge master-classify0428 into master feat: 预修改控制类算子迁移 Created-by: yelongjian Commit-by: 叶龙剑 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动,包括改动的原因和所采取的方法。--> 预修改控制类算子迁移 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。例如:关联Issue #000--> ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!4363 | 5 个月前 | |
修改doc tools低错 Co-authored-by: caiwenwen<caiwenwen6@h-partners.com> # message auto-generated for no-merge-commit merge: !4214 merge master into master 修改doc tools低错 Created-by: caiwenwen Commit-by: caiwenwen Merged-by: cann-robot Description: ## 描述 修改nn仓的markdown语法、htlm是否闭合、链接是否可以正常跳转、本地链接是否正常跳转修改低错修改 ## 关联的Issue 关联Issue [#1783](https://gitcode.com/cann/ops-nn/issues/1783) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 全部文档的链接跳转、htlm标签闭合、markdown语法规范问题 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [X] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!4214 | 5 个月前 | |
删除gawk依赖 Co-authored-by: chenqi317<chenqi317@huawei.com> # message auto-generated for no-merge-commit merge: !4152 merge master into master 删除gawk依赖 Created-by: chenqi317 Commit-by: chenqi317 Merged-by: cann-robot Description: ## 描述 ops-nn在编译过程中使用gawk工具用于打印时间戳,而gawk并非POSIX中原生支持,还需要用户额外安装; 解耦gawk工具,可以减少用户安装工具的数量,提升用户使用体验 一、代码使用到gawk的地方替换方式: main "$@" | gawk '{print strftime("[%Y-%m-%d %H:%M:%S]"), $0}' 替代为如下命令: main "$@" |while IFS= read -r line; do echo "$(date '+[%Y-%m-%d %H:%M:%S]') $line";done 二、资料中删除对gawk的依赖 三、install_deps.sh中删除安装gawk工具 <!--在这里详细描述你的改动,包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。例如:关联Issue #000--> https://gitcode.com/cann/ops-nn/issues/2269 https://gitcode.com/cann/ops-nn/issues/2259 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!4152 | 5 个月前 | |
[Docs] modify profile of quick_install Co-authored-by: wangzitao_leo<wangzitao4@huawei.com> # message auto-generated for no-merge-commit merge: !4156 merge profile_0423 into master [Docs] modify profile of quick_install Created-by: wangzitao_leo Commit-by: wangzitao_leo Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动,包括改动的原因和所采取的方法。--> 更新资料中: 1. --help时新增打印--soc支持的硬件平台; 2. quick_install指导资料中对于第三方依赖的安装资料调整,使得条理更加清晰,资料更为易用。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。例如:关联Issue #000--> Issue [#2274](https://gitcode.com/cann/ops-nn/issues/2274) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> 1. 根据资料安装第三方依赖 2. bash build.sh --help时打印--soc支持硬件平台列表 ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> 1. README.md 2. docs/zh/install/compile.md 3. docs/zh/install/quick_install.md 4. requirements.txt ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!4156 | 5 个月前 | |
refactor: 引用工程公共仓cann-cmake统一打包构建API,移除本地冗余脚本 Co-authored-by: yang-di52<yangdi52@huawei.com> # message auto-generated for no-merge-commit merge: !4109 merge bugfix1 into master refactor: 引用工程公共仓cann-cmake统一打包构建API,移除本地冗余脚本 Created-by: yang-di52 Commit-by: yang-di52 Merged-by: cann-robot Description: ## 描述 重构 ops-nn 仓库的打包构建流程,将本地维护的打包脚本和安装脚本替换为引用工程公共仓 cann-cmake 中统一的 API,降低构建工程的复杂度和维护成本。 ### 主要改动: 1. 升级 cann-cmake 版本:从 master-001 升级到 master-002,新增 SHA256 校验(fetch_cann_cmake.cmake) 2. 替换打包脚本引用路径:将 cmake/package.cmake 中所有 ${CMAKE_SOURCE_DIR}/scripts/package/ 和 ${CMAKE_SOURCE_DIR}/cmake/ 本地路径替换为 ${CANN_CMAKE_DIR}/ 公共路径 3. 移除本地冗余脚本:删除 cmake/makeself_built_in.cmake、cmake/makeself_custom.cmake、cmake/third_party/makeself-fetch.cmake 以及 scripts/package/common/ 和 scripts/package/latest_manager/ 下的全部 21 个文件 3. 更新打包配置:新增 CPACK_PACKAGE_PARAM_NAME 参数和 share_info_name XML 配置项(ops_math.xml) 4. 更新编译文档:同步更新 cann-cmake 版本号说明(docs/zh/install/compile.md) ## 关联的Issue [#2381](https://gitcode.com/cann/ops-nn/issues/2381) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 docs/zh/install/compile.md ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他,请描述:构建重构 See merge request: cann/ops-nn!4109 | 5 个月前 |
ops-nn
🔥Latest News
- [2026/01] 新增QuickStart,指导新手零基础入门算子项目部署(支持Docker环境)、算子开发和贡献流程。
- [2025/12] 开源算子支持Ascend 950PR/Ascend 950DT/KirinX90,可以通过CANN Simulator仿真工具开发调试;优化指南类文档,聚焦算子开发指南,明确最小交付件和关键示例代码,针对Ascend/samples仓算子提供迁移本项目的指导;新支持稀疏4:2量化matmul算子,针对稀疏矩阵使能硬件加速能力。
- [2025/11] 新支持算子index_fill、masked_scatter、scatter、tf_scatter_add、fused_cross_entropy_loss_with_max_sum。
- [2025/10] 新增experimental目录,完善贡献指南,支持开发者调试并贡献自定义算子。
- [2025/09] ops-nn项目首次上线,开源算子支持Atlas A2/A3系列产品。
🚀概述
ops-nn是CANN(Compute Architecture for Neural Networks)算子库中提供神经网络计算能力的高阶算子库,包括matmul类、activation类等算子,算子库架构图如下:

📌版本配套
本项目源码会跟随CANN软件版本发布,关于CANN软件版本与本项目标签的对应关系请参阅release仓库中的相应版本说明。 请注意,为确保您的源码定制开发顺利进行,请选择配套的CANN版本与Gitcode标签源码,使用master分支可能存在版本不匹配的风险。
🛠️环境准备
环境部署是体验本项目能力的前提,请先完成NPU驱动、CANN包安装等,确保环境正常。
⬇️源码下载
环境准备好后,下载与CANN版本配套的分支源码,命令如下,${tag_version}替换为分支标签名。
说明:若环境中已存在配套分支源码,可跳过本步骤,例如WebIDE默认已提供最新商发版CANN对应的源码。
git clone -b ${tag_version} https://gitcode.com/cann/ops-nn.git
说明:对于WebIDE环境,已默认提供最新商发CANN版本配套的源码,如需获取其他版本源码,参考上述命令获取。
📖学习教程
💬相关信息
PS:本项目功能和文档正在持续更新和完善中,欢迎您关注最新版本。