| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
补充4个算子rt2 infershape实现 + 9个算子常量折叠HOSTCPU改造 Co-authored-by: sujunwei3<sujunwei3@huawei.com> # message auto-generated for no-merge-commit merge: !4821 merge dev into master 补充4个算子rt2 infershape实现 + 9个算子常量折叠HOSTCPU改造 Created-by: sujunwei3 Commit-by: sujunwei3 Merged-by: cann-robot Description: ## 描述 本 PR 包含两部分改动： ### 1. 补充 4 个算子 rt2 infershape 实现 从 canndev 仓迁移 infershape 逻辑到 ops-math，包含 InferShape + InferDataType + InferShapeRange： - **coordinates_1d_to_2d**：rt1 改造为 rt2，3 个输出（row/col/n）的 shape+dtype+range 均来自输入 x - **split_d**：rt2 直接迁移，DYNAMIC_OUTPUT 按 split_dim/num_split 属性计算 shape+range - **slice_write**：rt1 改造为 rt2，输出 shape+dtype+range 来自输入 x，含 x/value dtype 一致性校验 - **calc_buckets_limit_and_offset**：rt1 改造为 rt2，2 个输出 shape+dtype 来自 bucket_list/ivf_offset - 同步迁移 UT 用例（共 8 个），改造为 InfershapeContextPara + ExecuteTestCase 格式 ### 2. 9 个算子常量折叠 HOSTCPU 改造 - CMakeLists.txt 加 HOSTCPU TRUE 参数 - 算子注册宏从 REGISTER_CPU_KERNEL 改为 OPS_MATH_REGISTER_CPU_KERNELV2 - 改造算子：calc_buckets_limit_and_offset、cast、coordinates_1d_to_2d、cumsum、lin_space、slice_write、split_v、topk、tile_with_axis - 编译产出 libopconstant_folding_math.so，strings 验证全部算子符号存在 ## 关联的Issue #2717 ## 测试 ### infershape UT 验证 - 执行 bash build.sh -u --ophost --soc=ascend910b --ops=<op_name> -j16 - 验证条件：exit code == 0 且日志含 [ PASSED ] N tests. - 8 个 UT 用例全部通过（含 dtype 不一致失败用例） ### 常量折叠编译验证 - 执行 bash build.sh --pkg --jit -j16 - 验证 libopconstant_folding_math.so 产出 - strings 验证 9 个算子符号均在 so 中 - RunCpuKernelV2 / GetCpuKernelV2 入口符号存在 ### 文档更新验证 - op_list.md 中 4 个算子 op_host 列 ×→√ - 2 个算子 op_kernel 列 ×→√（coordinates_1d_to_2d、slice_write） ## 文档更新 - docs/zh/op_list.md：更新 coordinates_1d_to_2d、split_d、slice_write、calc_buckets_limit_and_offset 的 op_host/op_kernel 标记 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!4821 | 11 天前 |