本项目是CANN提供的神经网络类计算算子库,实现网络在NPU上加速计算。
| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
fix_pull_request Co-authored-by: huangkejie1647<huangkejie3@huawei.com> # message auto-generated for no-merge-commit merge: !1219 merge fix_pull_request into master fix_pull_request Created-by: huangkejie1647 Commit-by: huangkejie1647 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动,包括改动的原因和所采取的方法。--> 修改pr描述模板,方便后续开发 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。例如:关联Issue #000--> <!-- 如果这个PR是为了解决特定的问题单,请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> 不涉及 ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> 不涉及 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ✅] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!1219 | 7 个月前 | |
delete duplicate file glu_tiling_kernel.hpp Co-authored-by: ni-zhihao<nizhihao@h-partners.com> # message auto-generated for no-merge-commit merge: !2979 merge master into master delete duplicate file glu_tiling_kernel.hpp Created-by: ni-zhihao Commit-by: ni-zhihao Merged-by: cann-robot Description: ## 描述 删除重复的glu_tiling_kernel.hpp文件 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。例如:关联Issue #000--> ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!2979 | 5 个月前 | |
移动version.cmake 文件 Co-authored-by: yang-di52<yangdi52@huawei.com> # message auto-generated for no-merge-commit merge: !3093 merge new into master 移动version.cmake 文件 Created-by: yang-di52 Commit-by: yang-di52 Merged-by: cann-robot Description: ## 描述 移动version.cmake 文件 ## 关联的Issue [#1671](https://gitcode.com/cann/ops-nn/issues/1671) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他,请描述:工程重构 See merge request: cann/ops-nn!3093 | 5 个月前 | |
simt子函数整改:根据当前编程规范将simt中调用的子函数加上__simt_callee__ 前缀 Co-authored-by: 李鑫<lixin433@huawei.com> # message auto-generated for no-merge-commit merge: !3011 merge simt into master simt子函数整改:根据当前编程规范将simt中调用的子函数加上__simt_callee__ 前缀 Created-by: lixin433 Commit-by: 李鑫 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动,包括改动的原因和所采取的方法。--> 根据当前编程规范将simt中调用的子函数加上__simt_callee__ 前缀 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。例如:关联Issue #000--> 关联Issue [#1661](https://gitcode.com/cann/ops-nn/issues/1661) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> 本地编译、验证均通过,二级冒烟已通过。 ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> 不涉及 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!3011 | 5 个月前 | |
noaicpu option is supported Co-authored-by: zihan0007<mazihan1@huawei.com> # message auto-generated for no-merge-commit merge: !1753 merge master into master noaicpu option is supported Created-by: zihan0007 Commit-by: zihan0007 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动,包括改动的原因和所采取的方法。--> 问题背景:不需要用到毕昇编译器的场景,因为构建时并没有安装毕昇编译器的任何东西,但aicpu因为涉及device侧运行,需要用到毕昇的交叉工具链,所以会报错工具链找不到。 解决方案:用--noaicpu参数选项,在构建aicpu算子的时候进行kernel隔离。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。例如:关联Issue #000--> 关联Issue [#1015](https://gitcode.com/cann/ops-nn/issues/1015) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> 线上ci编译验证 ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> 不涉及 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!1753 | 6 个月前 | |
conv2dbackward 转mm问题修复,NpuArch::DAV_2201不支持转mm Co-authored-by: huyihang<huyihang6@h-partners.com> # message auto-generated for no-merge-commit merge: !2990 merge convmm into master conv2dbackward 转mm问题修复,NpuArch::DAV_2201不支持转mm Created-by: huyihang1117 Commit-by: huyihang Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动,包括改动的原因和所采取的方法。--> 修复卷积反向1dcase,fmap=kernel场景下 在NpuArch::DAV_2201下运行失败 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。例如:关联Issue #000--> https://gitcode.com/cann/ops-nn/issues/1669 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> 失败的case,在新的代码添加之后运行成功 ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!2990 | 5 个月前 | |
添加AddRmsNormDynamicMxQuant算子实现 Co-authored-by: liiu-qii<liuqi330@huawei.com> # message auto-generated for no-merge-commit merge: !3020 merge ARNDfinal into master 添加AddRmsNormDynamicMxQuant算子实现 Created-by: duchaune Commit-by: liiu-qii Merged-by: cann-robot Description: ## 描述 AddRmsNormDynamicMxQuant Ascend950开发,具体包括: 1、算子原型定义、信息库注册和infershape功能; 2、tiling策略和kernel实现,包括R全载模板(mxfp8类型支持); 3、针对Ascend 950平台的编译配置与二进制生成规则; 4、新增tiling ut、kernel ut和infershape ut; 5、新增READEME文档和op_list文档更新。 ## 关联的Issue https://gitcode.com/cann/ops-nn/issues/1637 ## 测试 本地泛化精度和性能测试通过。 ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> 1、新增AddRmsNormDynamicMxQuant的算子README.md文件; 2、更新docs/zh/op_list.md和docs/zh/op_api_list.md,在相应位置加上AddRmsNormDynamicMxQuant算子描述。 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!3020 | 5 个月前 | |
improve QuickStart docs Co-authored-by: ZitaoWang<wangzitao4@huawei.com> # message auto-generated for no-merge-commit merge: !2707 merge improve_quickstart into master improve QuickStart docs Created-by: wangzitao_leo Commit-by: ZitaoWang Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动,包括改动的原因和所采取的方法。--> 简化QuickStart中环境部署章节,保留关键能力项操作;同时新增文档install目录 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。例如:关联Issue #000--> 关联Issue [#1493](https://gitcode.com/cann/ops-nn/issues/1493) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> NA ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> README、QuickStart、环境部署 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!2707 | 5 个月前 | |
[aicore]整改 Co-authored-by: yang-di52<yangdi52@huawei.com> # message auto-generated for no-merge-commit merge: !2261 merge master into master [aicore]整改 Created-by: yang-di52 Commit-by: yang-di52 Merged-by: cann-robot Description: ## 描述 编译器内部已经定义了__aicore__,算子侧需要去掉这个标识 ## 关联的Issue [#1276](https://gitcode.com/cann/ops-nn/issues/1276) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!2261 | 6 个月前 | |
md文档和示例代码修复 Co-authored-by: Chen_HaoWen<chenhaowen12@huawei.com> # message auto-generated for no-merge-commit merge: !2978 merge master_qbn into master md文档和示例代码修复 Created-by: Chen_HaoWen Commit-by: Chen_HaoWen Merged-by: cann-robot Description: ## 描述 md文档和示例代码修复 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。例如:关联Issue #000--> ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!2978 | 5 个月前 | |
refactor common headers for op_host/op_api Co-authored-by: 丛吉钰<congjiyu@h-partners.com> # message auto-generated for no-merge-commit merge: !1807 merge master into master refactor common headers for op_host/op_api Created-by: cong-jiyu Commit-by: 丛吉钰 Merged-by: cann-robot Description: ## 描述 迁移common目录下的文件到op_api/op_host中 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。例如:关联Issue #000--> ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!1807 | 6 个月前 | |
simt子函数整改:根据当前编程规范将simt中调用的子函数加上__simt_callee__ 前缀 Co-authored-by: 李鑫<lixin433@huawei.com> # message auto-generated for no-merge-commit merge: !3011 merge simt into master simt子函数整改:根据当前编程规范将simt中调用的子函数加上__simt_callee__ 前缀 Created-by: lixin433 Commit-by: 李鑫 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动,包括改动的原因和所采取的方法。--> 根据当前编程规范将simt中调用的子函数加上__simt_callee__ 前缀 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。例如:关联Issue #000--> 关联Issue [#1661](https://gitcode.com/cann/ops-nn/issues/1661) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> 本地编译、验证均通过,二级冒烟已通过。 ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> 不涉及 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!3011 | 5 个月前 | |
simt子函数整改:根据当前编程规范将simt中调用的子函数加上__simt_callee__ 前缀 Co-authored-by: 李鑫<lixin433@huawei.com> # message auto-generated for no-merge-commit merge: !3011 merge simt into master simt子函数整改:根据当前编程规范将simt中调用的子函数加上__simt_callee__ 前缀 Created-by: lixin433 Commit-by: 李鑫 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动,包括改动的原因和所采取的方法。--> 根据当前编程规范将simt中调用的子函数加上__simt_callee__ 前缀 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。例如:关联Issue #000--> 关联Issue [#1661](https://gitcode.com/cann/ops-nn/issues/1661) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> 本地编译、验证均通过,二级冒烟已通过。 ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> 不涉及 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!3011 | 5 个月前 | |
transdata后融合,回退n非对齐场景 Co-authored-by: z00939187<zhengyuhao3@huawei.com> # message auto-generated for no-merge-commit merge: !3047 merge master-resert into master transdata后融合,回退n非对齐场景 Created-by: zhengyuhao3 Commit-by: z00939187 Merged-by: cann-robot Description: ## 描述 transdata后融合内存优化,非对齐场景切换通路后导致网络用例性能下降,回退n非对齐场景 ## 关联的Issue https://gitcode.com/cann/ops-nn/issues/1662 ## 测试 二级冒烟 ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!3047 | 5 个月前 | |
raii examples and md修改 | 5 个月前 | |
获取整型属性潜在风险整改 Co-authored-by: zhuzemao<zhuzemao1@h-partners.com> # message auto-generated for no-merge-commit merge: !2712 merge attr_modify into master 获取整型属性潜在风险整改 Created-by: zhuzemao Commit-by: zhuzemao Merged-by: cann-robot Description: ## 描述 获取整型属性潜在风险整改 ## 关联的Issue https://gitcode.com/cann/ops-nn/issues/1614 ## 测试 不涉及 ## 文档更新 不涉及 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他,请描述:风险整改 See merge request: cann/ops-nn!2712 | 5 个月前 | |
池化算子和applyTopKTopP算子文档资料描述修正 Co-authored-by: chenfeng61<chenfeng61@huawei.com> # message auto-generated for no-merge-commit merge: !2972 merge fix/readme_cleancode into master 池化算子和applyTopKTopP算子文档资料描述修正 Created-by: chenyanbin8 Commit-by: chenfeng61 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动,包括改动的原因和所采取的方法。--> 利用AI文档扫描工具对仓内资料进行检查扫描,池化算子和applyTopKTopP算子存在资料描述不正确地方,包含①example示例代码使用printf函数建议显示包含“cstdio”头文件、②example代码中使用错误数据类型创建tensor数据类型错误/未校验、③错别字描述错误、④错误逻辑描述、⑤apply_top_k_top_p_with_sorted算子《README.md》中sorted_indices数据类型描述错误。 具体的错误描述位于issue中,对对应资料错误之处修正。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。例如:关联Issue #000--> ## 测试 <!--描述进行了哪些测试来验证你的改动。--> 不涉及功能代码,example示例代码运行成功 ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> aclnnAdaptiveAvgPool2d.md、aclnnAdaptiveAvgPool3d.md、aclnnAdaptiveAvgPool2dBackward.md、 aclnnAdaptiveAvgPool3dBackward.md、aclnnAdaptiveMaxPool2d.md、aclnnAdaptiveMaxPool3d.md、 aclnnAdaptiveMaxPool2dBackward.md、aclnnAdaptiveMaxPool2dBackward.md、aclnnAvgPool2d.md、 aclnnAvgPool3d.md、aclnnAvgPool2dBackward.md、aclnnAvgPool3dBackward.md、 aclnnMaxPool2dWithIndicesBackward.md、aclnnMaxPool2dWithMaskBackward.md、aclnnMaxPool3dWithArgmaxBackward.md、 aclnnMaxPool2dWithIndices.md、aclnnMaxPool2dWithMask.md、aclnnMaxPool3dWithArgmax.md、aclnnMaxPool.md、 apply_top_k_top_p_with_sorted算子README.md文档 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!2972 | 5 个月前 | |
add AscendAntiQuantV2 Co-authored-by: wang-siyang3<wangsiyang3@huawei.com> # message auto-generated for no-merge-commit merge: !1643 merge AscendAntiQuantV2 into master add AscendAntiQuantV2 Created-by: clwsy1 Commit-by: wang-siyang3 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动,包括改动的原因和所采取的方法。--> 1、AscendAntiQuantV2算子新增支持hifp8和fp8类型 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。例如:关联Issue #000--> 关联Issue [#1631](https://gitcode.com/cann/ops-nn/issues/1631) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> 已自测通过 ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> 涉及README.md文件和aclnn资料修改 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!1643 | 5 个月前 | |
Fix docs ans license issues of lstm_cell Co-authored-by: jisihuai<jisihuai@h-partners.com> # message auto-generated for no-merge-commit merge: !3015 merge test into master Fix docs ans license issues of lstm_cell Created-by: jisihuai Commit-by: jisihuai Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动,包括改动的原因和所采取的方法。--> 修复thnn_fused_lstm_cell、lstm算子资料中的相关问题与license问题 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。例如:关联Issue #000--> [#1125](https://gitcode.com/cann/ops-nn/issues/1125) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> rnn/thnn_fused_lstm_cell/docs/aclnnThnnFusedLstmCell.md rnn/thnn_fused_lstm_cell/README.md rnn/dynamic_rnn/docs/aclnnLSTM.md docs/zh/op_list.md docs/zh/op_api_list.md ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!3015 | 5 个月前 | |
添加AddRmsNormDynamicMxQuant算子实现 Co-authored-by: liiu-qii<liuqi330@huawei.com> # message auto-generated for no-merge-commit merge: !3020 merge ARNDfinal into master 添加AddRmsNormDynamicMxQuant算子实现 Created-by: duchaune Commit-by: liiu-qii Merged-by: cann-robot Description: ## 描述 AddRmsNormDynamicMxQuant Ascend950开发,具体包括: 1、算子原型定义、信息库注册和infershape功能; 2、tiling策略和kernel实现,包括R全载模板(mxfp8类型支持); 3、针对Ascend 950平台的编译配置与二进制生成规则; 4、新增tiling ut、kernel ut和infershape ut; 5、新增READEME文档和op_list文档更新。 ## 关联的Issue https://gitcode.com/cann/ops-nn/issues/1637 ## 测试 本地泛化精度和性能测试通过。 ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> 1、新增AddRmsNormDynamicMxQuant的算子README.md文件; 2、更新docs/zh/op_list.md和docs/zh/op_api_list.md,在相应位置加上AddRmsNormDynamicMxQuant算子描述。 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!3020 | 5 个月前 | |
BroadcastGradientArgs/LayerNorm支持下一代芯片 Co-authored-by: cai-chengchao<caichengchao1@h-partners.com> # message auto-generated for no-merge-commit merge: !2099 merge add_ops_0225 into master BroadcastGradientArgs/LayerNorm支持下一代芯片 Created-by: cai-chengchao Commit-by: cai-chengchao Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动,包括改动的原因和所采取的方法。--> BroadcastGradientArgs、LayerNorm算子支持下一代芯片 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。例如:关联Issue #000--> https://gitcode.com/cann/ops-nn/issues/1382 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ut、st、二级冒烟均测试通过 ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> 更新了norm/layer_norm/README.md 更新了index/broadcast_gradient_args/README.md ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他,请描述:BroadcastGradientArgs、LayerNorm算子支持下一代芯片 See merge request: cann/ops-nn!2099 | 5 个月前 | |
资料整改第一部分 Co-authored-by: Dukexin004<dukexin@h-partners.com> # message auto-generated for no-merge-commit merge: !2337 merge master into master 资料整改第一部分 Created-by: Dukexin004 Commit-by: Dukexin004 Merged-by: cann-robot Description: ## 描述 算子代码中支持的数据类型和README.md文件中不一致;将算子代码中不支持而README.md文件中支持的数据类型在参数说明中添加具体描述。 ## 关联的Issue https://gitcode.com/cann/ops-nn/issues/1315 ## 测试 ## 文档更新 更新了README.md文件。 ## 类型标签 - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!2337 | 6 个月前 | |
Format update: Adjusted the license description format in all relevant files. Co-authored-by: jiangqi<jiangqi47@h-partners.com> # message auto-generated for no-merge-commit merge: !293 merge license into master Format update: Adjusted the license description format in all relevant files. Created-by: jiangqi2025 Commit-by: jiangqi Merged-by: turing_project1 Description: ## 描述 <!--在这里详细描述你的改动,包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。例如:关联Issue #123--> <!-- 如果这个PR是为了解决特定的问题单,请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!293 | 9 个月前 | |
local exec CI build Co-authored-by: 丛吉钰<congjiyu@h-partners.com> # message auto-generated for no-merge-commit merge: !2247 merge master into master local exec CI build Created-by: cong-jiyu Commit-by: 丛吉钰 Merged-by: cann-robot Description: ## 描述 本地执行CI命令 ## 关联的Issue [#1261](https://gitcode.com/cann/ops-nn/issues/1261) <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。例如:关联Issue #000--> ## 测试 本地修改算子文件后构建 ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!2247 | 6 个月前 | |
release CHANGELOG.md Co-authored-by: ZitaoWang<wangzitao4@huawei.com> # message auto-generated for no-merge-commit merge: !921 merge changelog into master release CHANGELOG.md Created-by: wangzitao_leo Commit-by: ZitaoWang Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动,包括改动的原因和所采取的方法。--> 1. add CHANGELOG.md to describe changes of new released version of CANN. ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。例如:关联Issue #000--> <!-- 如果这个PR是为了解决特定的问题单,请在这里描述问题单单号。--> 关联Issue #512 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> NA ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> CHANGELOG.md ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!921 | 7 个月前 | |
移动version.cmake 文件 Co-authored-by: yang-di52<yangdi52@huawei.com> # message auto-generated for no-merge-commit merge: !3093 merge new into master 移动version.cmake 文件 Created-by: yang-di52 Commit-by: yang-di52 Merged-by: cann-robot Description: ## 描述 移动version.cmake 文件 ## 关联的Issue [#1671](https://gitcode.com/cann/ops-nn/issues/1671) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他,请描述:工程重构 See merge request: cann/ops-nn!3093 | 5 个月前 | |
adjust quick install md Co-authored-by: gitcode-chenjiao<chenjiao31@huawei.com> # message auto-generated for no-merge-commit merge: !1804 merge master into master adjust quick install md Created-by: gitcode-chenjiao Commit-by: gitcode-chenjiao Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动,包括改动的原因和所采取的方法。--> 完善环境部署中手动安装CANN和源码下载章节 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。例如:关联Issue #000--> 不涉及 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> 已经过评审 ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> quick_install.md QUICKSTART.md ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!1804 | 6 个月前 | |
init | 11 个月前 | |
修改OAT.xml 优化license匹配规则 Co-authored-by: yang-di52<yangdi52@huawei.com> # message auto-generated for no-merge-commit merge: !1942 merge master into master 修改OAT.xml 优化license匹配规则 Created-by: yang-di52 Commit-by: yang-di52 Merged-by: cann-robot Description: ## 描述 修改OAT.xml 优化license匹配规则 ## 关联的Issue [#691](https://gitcode.com/cann/ops-nn/issues/691) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他,请描述:优化工程 See merge request: cann/ops-nn!1942 | 6 个月前 | |
Reorganize the installation md Co-authored-by: yangyang016<yangyang306@hisilicon.com> # message auto-generated for no-merge-commit merge: !3017 merge master into master Reorganize the installation md Created-by: yangyang016 Commit-by: yangyang016 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动,包括改动的原因和所采取的方法。--> 重新调整环境部署的大纲逻辑 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。例如:关联Issue #000--> [#1624](https://gitcode.com/cann/ops-nn/issues/1624) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> README.md docs/QUICKSTART.md docs/zh/install/dir_structure.md docs/zh/install/quick_install.md docs/zh/invocation/quick_op_invocation.md ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!3017 | 5 个月前 | |
【描 述】 fix bug for eigen 5.0.0's line endings Co-authored-by: j00427146<jialimin1@huawei.com> # message auto-generated for no-merge-commit merge: !2026 merge master into master 【描 述】 fix bug for eigen 5.0.0's line endings Created-by: jialimin1 Commit-by: j00427146 Merged-by: cann-robot Description: ## 描述 解决eigen5.0.0版本中debug/msvc/.dat和debug/msvc/.natvis文件的换行符校验问题。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。例如:关联Issue #000--> ## 测试 [2026-02-25 10:32:35] About to compress 185236 KB of data... [2026-02-25 10:32:35] Adding files to archive named "cann-910b-ops-nn_9.0.0_linux-x86_64.run"... [2026-02-25 10:32:58] [INFO] exe_task: end to build kernel: add_layer_norm_binary_high_performance.json_6 --exe_time=2317 [2026-02-25 10:32:58] /usr/bin/make [2026-02-25 10:33:00] CRC: 2109050736 [2026-02-25 10:33:02] SHA256: 32f4e90af4109947954a4bf33a75f132bfd1f0f81c6172098b69140d4ef9c410 [2026-02-25 10:33:02] Skipping md5sum at user request [2026-02-25 10:33:02] [2026-02-25 10:33:03] Self-extractable archive "cann-910b-ops-nn_9.0.0_linux-x86_64.run" successfully created. [2026-02-25 10:33:03] build pkg success: /home/j00427146/cann-eco/ops-nn/build_out/cann-910b-ops-nn_9.0.0_linux-x86_64.run ## 文档更新 SECURITY.md ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!2026 | 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 | 6 个月前 | |
【描 述】 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 | 6 个月前 | |
fix:gen opc bug Co-authored-by: 丛吉钰<congjiyu@h-partners.com> # message auto-generated for no-merge-commit merge: !2953 merge master into master fix:gen opc bug Created-by: cong-jiyu Commit-by: 丛吉钰 Merged-by: cann-robot Description: ## 描述 修复生成opc的bug、编译不同算子时的bug、修复配置文件、修改编译日志:减少无用日志打印。 ## 关联的Issue [#1605](https://gitcode.com/cann/ops-nn/issues/1605) ## 测试 本地编译正常。 ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!2953 | 5 个月前 | |
rms_norm_dynamic_mx_quant支持fp8全载模板 Co-authored-by: 朱寒寒<zhuhanhan@huawei.com> # message auto-generated for no-merge-commit merge: !2894 merge rmsNormDynamicMxQuantF into master rms_norm_dynamic_mx_quant支持fp8全载模板 Created-by: zzz2048 Commit-by: 朱寒寒 Merged-by: cann-robot Description: ## 描述 RmsNormDynamicMxQuant Ascend950开发,具体包括: 1、算子原型定义、信息库注册和infershape功能; 2、tiling策略和kernel实现,包括R全载模板(mxfp8类型支持); 3、针对Ascend 950平台的编译配置与二进制生成规则; 4、新增tiling ut、kernel ut和infershape ut; 5、新增READEME文档和op_list文档更新。 ## 关联的Issue https://gitcode.com/cann/ops-nn/issues/1649 ## 测试 本地泛化精度和性能测试通过。 ## 文档更新 1、新增RmsNormDynamicMxQuant的README.md文件; 2、更新docs/zh/op_list.md,在相应位置加上RmsNormDynamicMxQuant算子描述。 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!2894 | 5 个月前 | |
添加patch工具依赖 Co-authored-by: yangyang016<yangyang306@hisilicon.com> # message auto-generated for no-merge-commit merge: !2531 merge offline_build_docs into master 添加patch工具依赖 Created-by: yangyang016 Commit-by: yangyang016 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动,包括改动的原因和所采取的方法。--> 本次PR旨在解决项目在编译或运行过程中因缺少patch工具依赖而导致构建失败的问题。通过补充依赖说明和增强自动化安装脚本,确保用户环境具备所有必要的前置条件,提升项目构建的成功率和易用性。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。例如:关联Issue #000--> ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!2531 | 5 个月前 | |
Support the fusionpass es project Co-authored-by: chenyifeng<chenyifeng27@h-partners.com> # message auto-generated for no-merge-commit merge: !907 merge pass_cmake_0121 into master Support the fusionpass es project Created-by: chen_0715 Commit-by: chenyifeng Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动,包括改动的原因和所采取的方法。--> 新增支持FusionPass编译的工程能力,以及UT测试的功能。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。例如:关联Issue #000--> <!-- 如果这个PR是为了解决特定的问题单,请在这里描述问题单单号。--> https://gitcode.com/cann/ops-nn/issues/540 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!907 | 7 个月前 | |
移动version.cmake 文件 Co-authored-by: yang-di52<yangdi52@huawei.com> # message auto-generated for no-merge-commit merge: !3093 merge new into master 移动version.cmake 文件 Created-by: yang-di52 Commit-by: yang-di52 Merged-by: cann-robot Description: ## 描述 移动version.cmake 文件 ## 关联的Issue [#1671](https://gitcode.com/cann/ops-nn/issues/1671) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他,请描述:工程重构 See merge request: cann/ops-nn!3093 | 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分支可能存在版本不匹配的风险。
⚡️快速入门
若您希望从零到一快速体验项目能力,请访问下述简易教程。
- 环境部署:本文是QuickStart和各类教程的操作前提,请先完成基础环境搭建和源码下载。
- QuickStart:针对源码编译、算子调用/开发/调试等关键能力,提供快速上手的简易指南。
📖学习教程
若您已学习完快速入门章节,对本项目有一定认知,并希望深入了解和体验项目,请访问下述详细教程。
- 算子列表:提供全量算子信息,方便您查看算子分类和功能。
- 算子调用:提供多种源码编译和执行算子样例(包括执行UT)的方法。
- 算子开发:提供算子端到端开发指南,从零学习创建算子工程、实现Tiling和Kernel核心交付件。
- 算子调试调优:提供常见算子调试和调优方法,如DumpTensor、msProf、Simulator等。
除了上述关键教程,还有其他文档介绍,例如算子调用方式、build参数说明、术语概念等,全量文档请访问docs。
🔍目录结构
关键目录如下,详细目录介绍参见项目目录。
├── activation # activation类算子
├── cmake # 项目工程编译目录
├── common # 项目公共头文件和公共源码
├── control # control类算子
├── conv # conv类算子
├── docs # 项目文档介绍
├── examples # 端到端算子开发和调用示例
├── experimental # 用户自定义算子存放目录
├── foreach # foreach类算子
├── index # index类算子
├── loss # loss类算子
├── matmul # matmul类算子
│ ├── transpose_batch_mat_mul # transpose_batch_mat_mul算子所有交付件,如Tiling、Kernel等
│ │ ├── docs # 算子说明文档
│ │ ├── examples # 算子使用示例
│ │ ├── op_graph # 算子构图相关目录
│ │ ├── op_host # 算子信息库、Tiling、InferShape相关实现目录
│ │ ├── op_api # 可选,算子aclnn接口实现目录,如未提供则表示此算子的aclnn接口会让工程自动生成
│ │ ├── op_kernel # 算子Kernel目录
│ │ ├── CMakeLists.txt # 算子编译配置文件
│ │ └── README.md # 算子介绍文档
│ ├── ...
│ └── CMakeLists.txt # 算子编译配置文件
├── ...
├── rnn # rnn类算子
├── scripts # 脚本目录,包含自定义算子、Kernel构建相关配置文件
├── tests # 测试工程目录
├── vfusion # vfusion类算子
├── CMakeLists.txt
├── README.md
├── build.sh # 项目工程编译脚本
├── install_deps.sh # 安装依赖包脚本
└── requirements.txt # 项目需要的第三方依赖包
💬相关信息
🤝联系我们
本项目功能和文档正在持续更新和完善中,建议您关注最新版本。
-
问题反馈:通过GitCode【Issues】提交问题。
-
社区互动:通过GitCode【讨论】参与交流。
-
技术专栏:通过GitCode【Wiki】获取技术文章,如系列化教程、优秀实践等。
技术专题 样例 算子性能优化 MatMul算子性能优化实践与效果分析 算子性能优化 MatMul算子VCV性能优化实践与效果分析