已合并
[feat]add aicpu op: sqrt #2539
FengHaozhan创建于 4月28日
[feat]add aicpu op: sqrt #2539
已合并
Pull Request已成功合入, 合并人@CANN-robot
(感谢 FengHaozhan 的贡献)4月28日 创建了 pull request,commit 6a7a23cc
CANN-robot
4月28日 评论:
4月28日 评论:
Thanks for your pull-request.
The full list of commands accepted by me can be found at here。
You can get sig-info at here
PR Approval Progress
✅ Congratulations! All modules have met the lgtm and approve requirements.
Module Approval Details
| module | lgtm status | approve status |
|---|---|---|
| */*/README.md | ✅ 陈娇, 周奇龙, wangrui_ (3/2) | ✅ 陈娇 (1/1) |
| math/sqrt | ✅ 周奇龙, wangrui_ (2/2) | ✅ 周奇龙, wangrui_ (2/1) |
💡 Tip:
- Committer can comment
/approveor/lgtm- Commenting
/approveimplies both code review (lgtm) and intent to merge (approve)
CLA Signature Pass
FengHaozhan, thanks for your pull request. All authors of the commits have signed the CLA. 👍


4月28日 添加了label:cann-cla/yes
4月28日 将FelixTang7,zhou-qilong,gitee-duhuiping,wangrui_,su-yueming,rxtfeng,loov1,zhangzijie,jia0liang,songkai111,gitcode-chenjiao,zhangyuxiang0119,llimwang,gubaocheng设为评审人
4月28日 将FelixTang7,zhou-qilong,gitee-duhuiping,wangrui_,su-yueming,rxtfeng,loov1,zhangzijie,jia0liang,songkai111,gitcode-chenjiao,zhangyuxiang0119,llimwang,gubaocheng设为评审人
4月28日 将FelixTang7,zhou-qilong,gitee-duhuiping,wangrui_,su-yueming,rxtfeng,loov1,zhangzijie,songkai111,gitcode-chenjiao,zhangyuxiang0119,llimwang,gubaocheng设为审查人
4月28日 将FelixTang7,zhou-qilong,gitee-duhuiping,wangrui_,su-yueming,rxtfeng,loov1,zhangzijie,songkai111,gitcode-chenjiao,zhangyuxiang0119,llimwang,gubaocheng设为审查人
此处折叠了49条消息 查看更多
4月30日 添加了label:lgtm
4月30日 关闭了关联的issue
4月30日 合入了pull request
描述
新增 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
测试
根据代码变更,测试场景如下:
AICPU 单元测试
tests/ut/op_kernel_aicpu/test_sqrt.cpp中的 UT 用例图模式示例验证
examples/test_geir_sqrt.cpp算子信息验证
sqrt.json中的 dtype/format 配置是否与 kernel 实现一致文档更新
math/sqrt/README.md,补充 AICPU 算子说明类型标签