已合并
[feat]add aicpu op: sqrt #2539
FengHaozhan创建于 4月28日
[feat]add aicpu op: sqrt #2539
已合并
FengHaozhan创建于 4月28日
FengHaozhan
FengHaozhan
4月28日

描述

新增 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 算子说明

类型标签

likedislike
Pull Request已成功合入, 合并人@CANN-robot
(感谢 FengHaozhan 的贡献)
FengHaozhanFengHaozhan
4月28日 创建了 pull request,commit 6a7a23cc
CANN-robot
CANN-robot成员
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 /approve or /lgtm
  • Commenting /approve implies 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. 👍

likedislike
CANN-robotCANN-robot成员
4月28日 添加了label:cann-cla/yes
CANN-robotCANN-robot成员
4月28日 将FelixTang7,zhou-qilong,gitee-duhuiping,wangrui_,su-yueming,rxtfeng,loov1,zhangzijie,jia0liang,songkai111,gitcode-chenjiao,zhangyuxiang0119,llimwang,gubaocheng设为评审人
CANN-robotCANN-robot成员
4月28日 将FelixTang7,zhou-qilong,gitee-duhuiping,wangrui_,su-yueming,rxtfeng,loov1,zhangzijie,songkai111,gitcode-chenjiao,zhangyuxiang0119,llimwang,gubaocheng设为审查人
此处折叠了49条消息 查看更多
RuiWang_成员
4月30日 评论:

/lgtm
/approve

likedislike
CANN-robotCANN-robot成员
4月30日 添加了label:lgtm
CANN-robot
CANN-robot成员
4月30日 评论:

Review Guide

This pull-request passes review.
Committers who wrote a comment of /approve are: 陈娇, 周奇龙, wangrui_.
Reviewers who wrote a comment of /lgtm are: 陈娇, wangrui_, 周奇龙.

likedislike
CANN-robotCANN-robot成员
4月30日 关闭了关联的issue
CANN-robotCANN-robot成员
4月30日 合入了pull request