master

cann-robot常量折叠和infershape整改

c257cedfAdd on Aug 26History

Files	Last commit	Last update
select_v2_aicpu.cpp	常量折叠和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
select_v2_aicpu.h	Migrate selectV2 AICPU operator from canndev to ops-math repository Co-authored-by: littleWhite0625<shizhenzhen@huawei.com> # message auto-generated for no-merge-commit merge: !2613 merge master into master Migrate selectV2 AICPU operator from canndev to ops-math repository Created-by: littleWhite0625 Commit-by: littleWhite0625 Merged-by: cann-robot Description: ## 描述 新增selectV2算子的AI CPU kernel实现，扩展CANN算子库在AI CPU上的算子支持范围。 ## 关联的Issue https://gitcode.com/cann/ops-math/issues/1486 ## 测试 根据代码变更，测试场景如下： 算子构建测试 执行 cmake 配置和编译，验证 CMake 配置变更正确 算子加载测试 验证select_v2.json 接口注册成功 单元测试 UT: bash build.sh -u --opkernel_aicpu --ops=select_v2 --soc=ascend910b GE IR验证 ## 文档更新 docs/zh/op_list.md新增select_v2 aicpu算子支持说明 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!2613	6 months ago
select_v2_aicpu_def.cpp	常量折叠和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