| aicpu tile op supports host constant folding v2 interface.
Co-authored-by: liu-wei<lovline.liuwei@huawei.com>
# message auto-generated for no-merge-commit merge:
!3092 merge master into master
aicpu tile op supports host constant folding v2 interface.
Created-by: liu-wei
Commit-by: liu-wei
Merged-by: cann-robot
Description: ## 描述
为 math/tile AICPU 算子补充 host constant folding v2 接口支持,使 Tile 算子在 host 侧常量折叠场景下可使用 ops-math AICPU host kernel 注册路径。
主要改动:
- 在 math/tile/op_kernel_aicpu/tile_aicpu.cpp 中引入 aicpu/math_aicpu_register.h。
- 将 TileCheckCopySupported 中的 host 判断由仅 RUN_ON_HOST 扩展为 RUN_ON_HOST 或 OPS_MATH_AICPU_HOST_KERNEL,host kernel 编译场景下关闭 copy hook。
- 将 REGISTER_CPU_KERNEL(kTile, TileCpuKernel) 替换为 OPS_MATH_REGISTER_CPU_KERNELV2(kTile, TileCpuKernel),适配 host constant folding v2 注册接口。
## 关联的Issue
无关联 Issue。
## 测试
建议补充以下验证:
- 编译构建 math/tile AICPU 算子,确认新增注册头文件和 V2 注册宏可正常通过编译。
- 执行 Tile 算子 AICPU UT/冒烟测试,确认原有 Tile 功能不受影响。
- 验证 host constant folding v2 场景下 Tile 算子可正常注册并完成常量折叠。
## 文档更新
不涉及文档更新。
## 类型标签
<!-- [x] 表示选中 -->
- [ ] Bug修复
- [x] 新特性
- [ ] 性能优化
- [ ] 文档更新
- [ ] 其他,请描述:
See merge request: cann/ops-math!3092 | 1 天前 |