| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
根据markdownlint规范修正md写作问题和link失效 Co-authored-by: gitcode-chenjiao<chenjiao31@huawei.com> # message auto-generated for no-merge-commit merge: !6398 merge master into master 根据markdownlint规范修正md写作问题和link失效 Created-by: gitcode-chenjiao Commit-by: gitcode-chenjiao Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 根据markdownlint规范修正md写作问题和link失效 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> [#2272](https://gitcode.com/cann/ops-math/issues/2272) ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ok ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> 所有md ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!6398 | 3 天前 | |
feat: 补充部分算子缺失的 aclnn API 示例 Co-authored-by: zhangqijia<zhangqijia1@h-partners.com> # message auto-generated for no-merge-commit merge: !4575 merge feat/issue-2281-aclnn-examples into master feat: 补充部分算子缺失的 aclnn API 示例 Created-by: gcw_SUaZx3UQ Commit-by: zhangqijia Merged-by: cann-robot Description: ## 描述 补充 Issue #2281 中归属 ops-math 的 10 个算子、12 个 aclnn API 调用示例： - 新增 8 个缺失的 aclnn API 示例。 - 将已有的 4 个示例调整为 CI 可识别的 test_aclnn_*.cpp 命名，并修正 BatchNormStats 示例文件名。 - 根据 Cdist 已有 AscendC kernel 实现，完善其 README 的产品支持、功能公式、参数及约束说明。 - 更新相关算子 README 中的调用入口。 ## 关联的Issue https://gitcode.com/cann/ops-math/issues/2281 ## 测试 ## 文档更新 - 更新 10 个算子目录下的 README.md，补充 aclnn 示例链接及对应 API 文档入口。 - 重写 Cdist README，删除“仅包含 aclnn 接口”的过期说明，并按仓内算子格式补充产品支持、功能、参数和约束。 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!4575 | 1 个月前 | |
refactor: 为公共op_api头文件增加math后缀 Co-authored-by: clinglai0517<laichangling@huawei.com> # message auto-generated for no-merge-commit merge: !5839 merge refactor/math-prefix-common-headers into master refactor: 为公共op_api头文件增加math后缀 Created-by: clinglai0517 Commit-by: clinglai0517 Merged-by: cann-robot Description: ## 描述 ops-math/common/inc/op_api 中的 level2_base.h、level2_base_caculation.h、op_api_def.h 名称缺少仓库标识。在软件包合并交付或公共 include 目录集成时，存在被其他仓库同名文件覆盖，或因 include 路径顺序产生误引用的风险。 本 MR 参考同目录已有的 op_resource_math.h 命名，为三个公共头文件增加 _math 后缀，并通过兼容转发头分阶段迁移现有引用。 ## 修改内容 - 新增 level2_base_math.h、level2_base_caculation_math.h、op_api_def_math.h，作为带仓库标识的正式实现； - 原 level2_base.h、level2_base_caculation.h、op_api_def.h 保留为轻量兼容转发头，避免未迁移代码立即失效； - 新头文件之间的内部依赖统一使用 _math 后缀路径； - 更新 classify_rule.yaml，让 level2_base_math.h 继承原组件及 UT 覆盖率分类； - 将 89 处旧头文件引用迁移到新名称： - level2_base.h：20 处； - level2_base_caculation.h：9 处； - op_api_def.h：60 处； - 当前 MR 共修改 81 个文件，新增 434 行、删除 384 行。 为减少无关格式变化和后续合并冲突，本次未迁移会触发额外 clang-format、尾部空白或 EOF 修复，或组合触发既有 UT 隔离问题的文件。这些文件将拆分到后续 MR 逐步处理。 ## 剩余迁移 当前仓内仍有 67 处旧引用，分布在 65 个唯一文件中，继续通过兼容转发头工作： - level2_base.h：24 个文件； - level2_base_caculation.h：2 个文件； - op_api_def.h：41 个文件。 待旧引用分批迁移完成后，再结合实际打包和安装规则移除兼容转发头。本 MR 不移除旧头文件，也不声明已完成最终软件包去重。 ## 关联的 Issue 关联并在合入后关闭 #3243： https://gitcode.com/cann/ops-math/issues/3243 ## 测试 已完成： - 对本轮新增的 69 个纯引用替换文件逐行检查，80 组增删均仅修改目标 include，无其他代码或格式变化； - 额外纳入 aclnn_log10.cpp 和 aclnn_logdet.cpp，每个文件最终均为 3 行新增、3 行删除，低于 10 行修改阈值；math/sin/op_api/aclnn_sin.cpp 因聚合 UT 存在平台状态泄漏暂缓迁移； - 在 CANN 9.0.0 容器执行完整 pre-commit，尾部空白、EOF、clang-format、codespell 和 OAT 等检查全部通过； - 在 CANN 9.0.0 容器执行 bash build.sh --opapi -j32，全量构建通过； - 成功生成 build/libopapi_math.so，SHA-256：abaa8f8408979d931a8dad09f8e06875722fc42bbdddc72aaf9f14944bc163ea； - classify_rule.yaml 可正常解析，三个旧文件继续作为兼容转发头使用。 ## 文档更新 不涉及用户文档变更。 ## 类型标签 - [ ] Bug 修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他：代码重构 See merge request: cann/ops-math!5839 | 11 天前 | |
增加A2 A3新算子支持 Co-authored-by: slx2008<shaolixin@huawei.com> # message auto-generated for no-merge-commit merge: !448 merge master into master 增加A2 A3新算子支持 Created-by: songkai111 Commit-by: slx2008 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!448 | 9 个月前 | |
常量折叠和infershape整改 Co-authored-by: liu-lu<www.liulu824910939@qq.com> # message auto-generated for no-merge-commit merge: !4582 merge master into master 常量折叠和infershape整改 Created-by: liu-lu Commit-by: liu-lu Merged-by: cann-robot Description: ## 描述 本 PR 为 6 个 AICPU 算子（add、round、search_sorted、select_v2、square、squared_difference）使能 GE 图编译期常量折叠，并对 search_sorted、select_v2 进行 infershape/OpDef 整改。 ### 变更内容 **1. 常量折叠使能（6 算子）** | 算子 | CMakeLists | kernel 注册 | |------|-----------|-------------| | add | 增加 HOSTCPU TRUE | REGISTER_CPU_KERNEL → OPS_MATH_REGISTER_CPU_KERNELV2 | | round | 增加 HOSTCPU TRUE | REGISTER_CPU_KERNEL → OPS_MATH_REGISTER_CPU_KERNELV2 | | search_sorted | 增加 HOSTCPU TRUE | REGISTER_CPU_KERNEL → OPS_MATH_REGISTER_CPU_KERNELV2 | | select_v2 | 增加 HOSTCPU TRUE | REGISTER_CPU_KERNEL → OPS_MATH_REGISTER_CPU_KERNELV2 | | square | 增加 HOSTCPU TRUE | REGISTER_CPU_KERNEL → OPS_MATH_REGISTER_CPU_KERNELV2 | | squared_difference | 增加 HOSTCPU TRUE | REGISTER_CPU_KERNEL → OPS_MATH_REGISTER_CPU_KERNELV2 | - 各 kernel 增加 #include "aicpu/math_aicpu_register.h"； - CMakeLists 增加 HOSTCPU TRUE，将算子编入 host 侧折叠 so（libopconstant_folding_math.so）。 **2. infershape / OpDef 整改** - **search_sorted**：新增 op_host/search_sorted_infershape.cpp，实现 InferShape4SearchSorted，输出 shape 取 values（input1）的 shape； - **select_v2**：新增 op_kernel_aicpu/select_v2_aicpu_def.cpp（基于 OpDef 注册输入输出数据类型），删除旧 select_v2.json（JSON 配置方式），统一为 OpDef 注册链路。 ## 关联的Issue #2847 ## 测试 根据代码变更，测试场景如下： 1. **编译构建测试** - 执行 cmake 配置与编译，验证 6 个算子 CMakeLists HOSTCPU TRUE 配置变更正确； - 确认 libopconstant_folding_math.so 中已编入上述 6 个算子的 host 侧 kernel。 2. **常量折叠验证** - 输入为常量时，验证算子可在图编译期被折叠求值（host 侧执行），运行期不再下发 device kernel； - 输入为非常量时，验证 device 侧执行逻辑与折叠前保持一致。 3. **InferShape 验证（search_sorted）** - 验证 search_sorted 输出 shape 与 values 输入 shape 一致； - 覆盖多维度、边界 shape 场景。 4. **算子注册/加载验证（select_v2）** - 验证 select_v2 基于 OpDef 注册后接口可正确加载与调用； - 覆盖全数据类型（COMPLEX128/COMPLEX64/DOUBLE/FLOAT/FLOAT16/INT16/INT32/INT64/INT8/UINT16/UINT32/UINT64/UINT8/BOOL）。 5. **算子泛化** - 对 6 个算子执行二级冒烟与泛化用例，确认功能正确性不受影响。 ## 文档更新 本 PR 不涉及 README/aclnn 接口文档更新。 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [x] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!4582 | 1 个月前 | |
fix(aicpu): resolve code quality warnings Co-authored-by: pan-tong<pantong@huawei.com> # message auto-generated for no-merge-commit merge: !5876 merge fix/warning-handling into master fix(aicpu): resolve code quality warnings Created-by: pan-tong Commit-by: pan-tong Merged-by: cann-robot Description: ## 描述 修复 RightShift、SearchSorted、SelectV2、SquaredDifference 和 Tile AICPU 代码规范告警： - 为无状态成员函数补充 const，为重写析构函数补充 override。 - 补充显式类型转换，使用强类型枚举并显式传递宏依赖参数。 - 移除未使用头文件，不改变算子计算逻辑。 ## 关联的Issue 无 ## 测试 - bash build.sh -u --opkernel_aicpu --ops=right_shift,search_sorted,select_v2,squared_difference,tile - 91/91 PASSED ## 文档更新 无 ## 类型标签 - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他：代码规范告警整改 See merge request: cann/ops-math!5876 | 28 天前 | |
修复 aicpu ut case依赖 动态生成测试数据问题 Co-authored-by: qiang_zq<qiang.zhangqiang@huawei.com> # message auto-generated for no-merge-commit merge: !660 merge aicpu-ut into master 修复 aicpu ut case依赖 动态生成测试数据问题 Created-by: qiang_zq Commit-by: qiang_zq Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 部分Ai Cpu算子的UT用例，依赖python脚本动态生成测试数据，python脚本中依赖torch和tensorflow模块。用户开发算子时，通过社区获取并安装CANN包部署环境，并不会安装这2个py模块，因此依赖torch和tensorflow模块生成测试数据的UT用例，在出厂时是执行不成功的。针对这部分UT用例进行整改，使其不依赖未安装的torch和tensorflow模块，在出厂时默认可以执行成功。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> NA ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> bash build.sh -u --opkernel_aicpu  ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!660 | 8 个月前 | |
常量折叠和infershape整改 Co-authored-by: liu-lu<www.liulu824910939@qq.com> # message auto-generated for no-merge-commit merge: !4582 merge master into master 常量折叠和infershape整改 Created-by: liu-lu Commit-by: liu-lu Merged-by: cann-robot Description: ## 描述 本 PR 为 6 个 AICPU 算子（add、round、search_sorted、select_v2、square、squared_difference）使能 GE 图编译期常量折叠，并对 search_sorted、select_v2 进行 infershape/OpDef 整改。 ### 变更内容 **1. 常量折叠使能（6 算子）** | 算子 | CMakeLists | kernel 注册 | |------|-----------|-------------| | add | 增加 HOSTCPU TRUE | REGISTER_CPU_KERNEL → OPS_MATH_REGISTER_CPU_KERNELV2 | | round | 增加 HOSTCPU TRUE | REGISTER_CPU_KERNEL → OPS_MATH_REGISTER_CPU_KERNELV2 | | search_sorted | 增加 HOSTCPU TRUE | REGISTER_CPU_KERNEL → OPS_MATH_REGISTER_CPU_KERNELV2 | | select_v2 | 增加 HOSTCPU TRUE | REGISTER_CPU_KERNEL → OPS_MATH_REGISTER_CPU_KERNELV2 | | square | 增加 HOSTCPU TRUE | REGISTER_CPU_KERNEL → OPS_MATH_REGISTER_CPU_KERNELV2 | | squared_difference | 增加 HOSTCPU TRUE | REGISTER_CPU_KERNEL → OPS_MATH_REGISTER_CPU_KERNELV2 | - 各 kernel 增加 #include "aicpu/math_aicpu_register.h"； - CMakeLists 增加 HOSTCPU TRUE，将算子编入 host 侧折叠 so（libopconstant_folding_math.so）。 **2. infershape / OpDef 整改** - **search_sorted**：新增 op_host/search_sorted_infershape.cpp，实现 InferShape4SearchSorted，输出 shape 取 values（input1）的 shape； - **select_v2**：新增 op_kernel_aicpu/select_v2_aicpu_def.cpp（基于 OpDef 注册输入输出数据类型），删除旧 select_v2.json（JSON 配置方式），统一为 OpDef 注册链路。 ## 关联的Issue #2847 ## 测试 根据代码变更，测试场景如下： 1. **编译构建测试** - 执行 cmake 配置与编译，验证 6 个算子 CMakeLists HOSTCPU TRUE 配置变更正确； - 确认 libopconstant_folding_math.so 中已编入上述 6 个算子的 host 侧 kernel。 2. **常量折叠验证** - 输入为常量时，验证算子可在图编译期被折叠求值（host 侧执行），运行期不再下发 device kernel； - 输入为非常量时，验证 device 侧执行逻辑与折叠前保持一致。 3. **InferShape 验证（search_sorted）** - 验证 search_sorted 输出 shape 与 values 输入 shape 一致； - 覆盖多维度、边界 shape 场景。 4. **算子注册/加载验证（select_v2）** - 验证 select_v2 基于 OpDef 注册后接口可正确加载与调用； - 覆盖全数据类型（COMPLEX128/COMPLEX64/DOUBLE/FLOAT/FLOAT16/INT16/INT32/INT64/INT8/UINT16/UINT32/UINT64/UINT8/BOOL）。 5. **算子泛化** - 对 6 个算子执行二级冒烟与泛化用例，确认功能正确性不受影响。 ## 文档更新 本 PR 不涉及 README/aclnn 接口文档更新。 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [x] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!4582 | 1 个月前 | |
产品名变更 Co-authored-by: gitcode-chenjiao<chenjiao31@huawei.com> # message auto-generated for no-merge-commit merge: !5912 merge master into master 产品名变更 Created-by: gitcode-chenjiao Commit-by: gitcode-chenjiao Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> AI处理器对应的产品型号名称更新 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> [#3258](https://gitcode.com/cann/ops-math/issues/3258) ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ok ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> *.md ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!5912 | 27 天前 |
SearchSorted
产品支持情况
| 产品 | 是否支持 |
|---|---|
| Ascend 950PR&950DT系列产品 | √ |
| Atlas A3系列产品 | √ |
| Atlas A2系列产品 | √ |
| Atlas 200I/500 A2推理产品 | √ |
| Atlas推理系列产品 | √ |
| Atlas训练系列产品 | √ |
功能说明
- 算子功能：在一个已排序的张量sorted_sequence中查找给定张量values应该插入的位置。返回与values相同大小的张量，其中每个元素表示给定值在原始张量中应该插入的位置。
参数说明
| 参数名 | 输入/输出/属性 | 描述 | 数据类型 | 数据格式 |
|---|---|---|---|---|
| sorted_sequence | 输入 | 输入Tensor，其最后一维的数值按升序排列。 | FLOAT16、FLOAT、INT8、INT16、INT32、INT64、UINT8 | ND |
| values | 输入 | 待插入的Tensor，数据类型必须与输入sorted_sequence相同，除了最后一维外，其余维度必须与sorted_sequence相同。 | FLOAT16、FLOAT、INT8、INT16、INT32、INT64、UINT8 | ND |
| sorter | 可选输入 | 可选输入Tensor，其shape与未排序的sorted_sequence相同，包含将其在最内层维度升序排序的索引序列。 | INT64 | ND |
| dtype | 可选属性 |
• 指定输出Tensor的数据类型，仅支持INT64、INT32。 • 默认值为INT64。 |
TYPE | - |
| right | 可选属性 |
• 可选布尔值，false表示当序列中存在相同值、插入位置不唯一时，插入位置靠左对齐；true表示在这种情况下插入位置靠右对齐。 • 默认值为false。 |
BOOL | - |
| out | 输出 | 输出Tensor，shape与输入values相同。 | INT32、INT64 | ND |
约束说明
无
调用说明
| 调用方式 | 调用样例 | 说明 |
|---|---|---|
| aclnn调用 | test_aclnn_search_sorted | 通过aclnnSearchSorted接口方式调用SearchSorted算子。 |
| aclnn调用 | test_aclnn_search_sorteds | 通过aclnnSearchSorteds接口方式调用SearchSorted算子。 |
| 图模式调用 | test_geir_search_sorted | 通过算子IR构图方式调用SearchSorted算子。 |