| Files | Last commit | Last update |
|---|---|---|
master翻译合入 Co-authored-by: caiwenwen<caiwenwen6@h-partners.com> # message auto-generated for no-merge-commit merge: !6751 merge master into master master翻译合入 Created-by: caiwenwen Commit-by: caiwenwen Merged-by: cann-robot Description: ## 描述 合入math仓翻译的内容 ## 关联的Issue https://gitcode.com/cann/ops-math/issues/4170 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 docs/context，aclnn等商发文档 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!6751 | 14 days ago | |
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 | 2 months ago | |
fix: E*9****内部错误码整改为正式语义码并保留内部异常EZ9999上报 Co-authored-by: chenyifeng<chenyifeng27@h-partners.com> # message auto-generated for no-merge-commit merge: !6606 merge fix_errorNum_1021 into master fix: E*9****内部错误码整改为正式语义码并保留内部异常EZ9999上报 Created-by: chen_0715 Commit-by: chenyifeng Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 本次 PR 整改以下 43 个算子中非法使用的内部错误码（E*9****），使其符合错误码使用规范： - cholesky / complex / cross / cummax / cummin / cumprod / get_shape / histogram_v2 / is_finite / is_inf - kl_div_v2 / lin_space / log_add_exp / non_finite_check / pdist / polar / pows / radix_top_k - reduce_log_sum / reduce_log_sum_exp / reduce_max / reduce_mean / reduce_mean_with_count / reduce_min - reduce_nansum / reduce_std_with_mean / rfft1_d / search_sorted / signbit / sin / sinc / sinkhorn / topk - affine_grid / amp_update_scale / angle_v2 / arg_max_v2 / arg_min / arg_min_with_value / atan2 / bessel_i1e / cdist_grad - conversion/tril 改动原因：上述算子 op_host / op_api / op_graph 源码中存在裸 OP_LOGE （EZ9999 兜底上报）、 OP_CHECK_NULL_WITH_CONTEXT / OP_CHECK_NOTNULL （EZ9999 上报）等不符合错误码使用规范的上报路径，需按失败原因分类整改为正式语义码或内部透传。 采取的方法： - op_host 参数/输入校验类（shape / dtype / format / attr 值域等）裸 OP_LOGE → 正式语义宏（EZ0004 / EZ0008–EZ0038） - op_host 内部 API 失败类（平台能力查询、tiling API 失败等）→ OP_LOGE_WITHOUT_REPORT 透传，去掉 EZ9999 上报、保留日志 - OP_CHECK_NULL_WITH_CONTEXT （EZ9999）→ 显式空指针检查 + 正式码宏 - op_api OP_CHECK_NOTNULL （EZ9999）→ 显式检查 + OP_LOGE_WITHOUT_REPORT(ACLNN_ERR_INNER_NULLPTR) + 返回公共错误码 - op_api OP_LOGE(ACLNN_ERR_INNER_* / RUNTIME_ERROR) （561xxx / 361xxx）→ OP_LOGE_WITHOUT_REPORT ；正式参数码 ACLNN_ERR_PARAM_* （161xxx）保留 - 删除 CREATE_EXECUTOR 之后不可达的空指针死检查（ CHECK_RET(executor.get()!=nullptr, ACLNN_ERR_INNER_CREATE_EXECUTOR) ，40 处） - op_graph（histogram_v2 / reduce_mean 融合 pass）E9 上报 → 正式码或透传 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> https://gitcode.com/cann/ops-math/issues/3977 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> - 43 算子 op_host / op_api / op_graph 全量编译通过，exit 0（910b + 950/arch35） - 本地增加ut测试130 用例全量通过（910b 83 + 950 47），日志级验证触发 78 次正式 EZ 错误码上报，覆盖 14 个正式错误码 - 原版 math_op_host_ut 300 用例全量通过，无回归 - 静态复扫：范围内 E*9**** 正式上报清零（残留均为 CHECK_RET 返回公共码 / 动态 return 等允许项） - 附：适配 CANN 9.3.0 atvoss 库 tiling key 编码变化，更新 reduce_max / reduce_mean_with_count 测试的 9 处 expectedTilingKey （官方 master 在 9.3.0 下同样失败，非本 PR 引入） ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!6606 | 15 days ago | |
增加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 | 10 months ago | |
fix: E*9****内部错误码整改为正式语义码并保留内部异常EZ9999上报 Co-authored-by: chenyifeng<chenyifeng27@h-partners.com> # message auto-generated for no-merge-commit merge: !6606 merge fix_errorNum_1021 into master fix: E*9****内部错误码整改为正式语义码并保留内部异常EZ9999上报 Created-by: chen_0715 Commit-by: chenyifeng Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 本次 PR 整改以下 43 个算子中非法使用的内部错误码（E*9****），使其符合错误码使用规范： - cholesky / complex / cross / cummax / cummin / cumprod / get_shape / histogram_v2 / is_finite / is_inf - kl_div_v2 / lin_space / log_add_exp / non_finite_check / pdist / polar / pows / radix_top_k - reduce_log_sum / reduce_log_sum_exp / reduce_max / reduce_mean / reduce_mean_with_count / reduce_min - reduce_nansum / reduce_std_with_mean / rfft1_d / search_sorted / signbit / sin / sinc / sinkhorn / topk - affine_grid / amp_update_scale / angle_v2 / arg_max_v2 / arg_min / arg_min_with_value / atan2 / bessel_i1e / cdist_grad - conversion/tril 改动原因：上述算子 op_host / op_api / op_graph 源码中存在裸 OP_LOGE （EZ9999 兜底上报）、 OP_CHECK_NULL_WITH_CONTEXT / OP_CHECK_NOTNULL （EZ9999 上报）等不符合错误码使用规范的上报路径，需按失败原因分类整改为正式语义码或内部透传。 采取的方法： - op_host 参数/输入校验类（shape / dtype / format / attr 值域等）裸 OP_LOGE → 正式语义宏（EZ0004 / EZ0008–EZ0038） - op_host 内部 API 失败类（平台能力查询、tiling API 失败等）→ OP_LOGE_WITHOUT_REPORT 透传，去掉 EZ9999 上报、保留日志 - OP_CHECK_NULL_WITH_CONTEXT （EZ9999）→ 显式空指针检查 + 正式码宏 - op_api OP_CHECK_NOTNULL （EZ9999）→ 显式检查 + OP_LOGE_WITHOUT_REPORT(ACLNN_ERR_INNER_NULLPTR) + 返回公共错误码 - op_api OP_LOGE(ACLNN_ERR_INNER_* / RUNTIME_ERROR) （561xxx / 361xxx）→ OP_LOGE_WITHOUT_REPORT ；正式参数码 ACLNN_ERR_PARAM_* （161xxx）保留 - 删除 CREATE_EXECUTOR 之后不可达的空指针死检查（ CHECK_RET(executor.get()!=nullptr, ACLNN_ERR_INNER_CREATE_EXECUTOR) ，40 处） - op_graph（histogram_v2 / reduce_mean 融合 pass）E9 上报 → 正式码或透传 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> https://gitcode.com/cann/ops-math/issues/3977 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> - 43 算子 op_host / op_api / op_graph 全量编译通过，exit 0（910b + 950/arch35） - 本地增加ut测试130 用例全量通过（910b 83 + 950 47），日志级验证触发 78 次正式 EZ 错误码上报，覆盖 14 个正式错误码 - 原版 math_op_host_ut 300 用例全量通过，无回归 - 静态复扫：范围内 E*9**** 正式上报清零（残留均为 CHECK_RET 返回公共码 / 动态 return 等允许项） - 附：适配 CANN 9.3.0 atvoss 库 tiling key 编码变化，更新 reduce_max / reduce_mean_with_count 测试的 9 处 expectedTilingKey （官方 master 在 9.3.0 下同样失败，非本 PR 引入） ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!6606 | 15 days ago | |
[BugFix] 修复 SearchSorted 空 sorter 非法解引用 Co-authored-by: ayuemoon<15802951036@163.com> # message auto-generated for no-merge-commit merge: !6693 merge fix/search-sorted-empty-sorter-3863 into master [BugFix] 修复 SearchSorted 空 sorter 非法解引用 Created-by: ayuemoon Commit-by: ayuemoon Merged-by: cann-robot Description: ## 描述 SearchSorted AICPU kernel 在参数校验阶段直接读取 sorter 的最大值、最小值。当 sorted_sequence 和 sorter 均为空（例如形状为 {2, 0}）时，元素数一致性检查会放行，随后解引用空范围的结束迭代器，可能导致越界读取或空指针访问。 在 sorter 类型和元素数校验后，仅对非空 sorter 获取数据、检查数据指针并校验索引范围；空 sorter 跳过数据读取，继续执行输出及形状校验。补充空 sorter、后续参数校验、正常索引、负索引、越界索引及非空 sorter 数据指针为空的回归用例。测试文件按仓内 clang-format 配置统一格式。 ## 关联的Issue Fixes #3863 https://gitcode.com/cann/ops-math/issues/3863 ## 测试 - pre-commit（含格式及 OAT 检查）、git diff --check 通过。 - CANN 9.2.0 环境编译成功，SearchSorted AICPU UT **21/21 通过**。 - 未进行设备端验证。 ## 文档更新 本次修复参数校验中的空范围访问问题，无接口变更，无需更新文档。 ## 类型标签 - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!6693 | 14 days ago | |
[BugFix] 修复 SearchSorted 空 sorter 非法解引用 Co-authored-by: ayuemoon<15802951036@163.com> # message auto-generated for no-merge-commit merge: !6693 merge fix/search-sorted-empty-sorter-3863 into master [BugFix] 修复 SearchSorted 空 sorter 非法解引用 Created-by: ayuemoon Commit-by: ayuemoon Merged-by: cann-robot Description: ## 描述 SearchSorted AICPU kernel 在参数校验阶段直接读取 sorter 的最大值、最小值。当 sorted_sequence 和 sorter 均为空（例如形状为 {2, 0}）时，元素数一致性检查会放行，随后解引用空范围的结束迭代器，可能导致越界读取或空指针访问。 在 sorter 类型和元素数校验后，仅对非空 sorter 获取数据、检查数据指针并校验索引范围；空 sorter 跳过数据读取，继续执行输出及形状校验。补充空 sorter、后续参数校验、正常索引、负索引、越界索引及非空 sorter 数据指针为空的回归用例。测试文件按仓内 clang-format 配置统一格式。 ## 关联的Issue Fixes #3863 https://gitcode.com/cann/ops-math/issues/3863 ## 测试 - pre-commit（含格式及 OAT 检查）、git diff --check 通过。 - CANN 9.2.0 环境编译成功，SearchSorted AICPU UT **21/21 通过**。 - 未进行设备端验证。 ## 文档更新 本次修复参数校验中的空范围访问问题，无接口变更，无需更新文档。 ## 类型标签 - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!6693 | 14 days ago | |
常量折叠和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 | 2 months ago | |
产品名变更 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 | 1 month ago |
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算子。 |