| [feat]add aicpu op: sqrt
Co-authored-by: FengHaozhan<fenghaozhan@huawei.com>
# message auto-generated for no-merge-commit merge:
!2539 merge sqrt_aicpu into master
[feat]add aicpu op: sqrt
Created-by: FengHaozhan
Commit-by: FengHaozhan
Merged-by: cann-robot
Description: ## 描述
新增 AICPU 算子 sqrt,实现对张量各元素求平方根(√x)的功能,支持多种数据类型(float16、float32、float64、int32)及图模式(Graph Mode)调用。
主要变更:
- sqrt_aicpu.cpp / sqrt_aicpu.h:AICPU kernel 实现,基于模板化实现多数据类型支持
- sqrt.json:算子信息描述文件,定义输入输出的 dtype/format 列表
- tests/ut/op_kernel_aicpu/test_sqrt.cpp:覆盖各数据类型的单元测试用例
- examples/test_geir_sqrt.cpp:图模式端到端示例,验证算子在 GE 图中的运行
- math/sqrt/README.md:补充算子说明
## 关联的Issue
#1446
## 测试
根据代码变更,测试场景如下:
1. **AICPU 单元测试**
- 执行 tests/ut/op_kernel_aicpu/test_sqrt.cpp 中的 UT 用例
- 覆盖 float16、float32、float64、int32 数据类型,验证各类型计算结果正确
2. **图模式示例验证**
- 编译并运行 examples/test_geir_sqrt.cpp
- 验证 sqrt 算子在 GE 图模式下能正确加载、执行并输出预期结果
3. **算子信息验证**
- 检查 sqrt.json 中的 dtype/format 配置是否与 kernel 实现一致
## 文档更新
- 更新 math/sqrt/README.md,补充 AICPU 算子说明
## 类型标签
<!-- [x] 表示选中 -->
- [ ] Bug修复
- [x] 新特性
- [ ] 性能优化
- [ ] 文档更新
- [ ] 其他,请描述:
See merge request: cann/ops-math!2539 | 1 个月前 |