已合并
feat: 迁移 Conj 与 ConjugateTranspose AICPU 算子(含 infershape rt1→rt2) #3778
feat: 迁移 Conj 与 ConjugateTranspose AICPU 算子(含 infershape rt1→rt2) #3778
已合并
Ding_Jing创建于 7月3日
Ding_Jing
Ding_Jing成员
7月3日

描述

从 canndev 迁移两个 AICPU 算子到 ops-math,并将 infershape 由 rt1 适配为 rt2:

  • Conj → math/conj/(逐元素复数共轭)
  • ConjugateTranspose → conversion/conjugate_transpose/(按 perm 转置 + 共轭)

每算子交付:op_graph(proto + InferDataType)、op_host(infershape rt2)、op_kernel_aicpu(kernel + opdef)、examples(图模式)、tests/ut(infershape + kernel)、README;docs/zh/op_list.md 补充条目。

infershape 适配要点:

  • Conj 为 element-wise,复用 Ops::Base::InferShape4Elewise
  • ConjugateTranspose 按 perm 常量重排 + InputsDataDependency({perm});perm 校验为非负 [0, rank-1],与本算子 AICPU kernel、canndev rt1、TF 语义一致(format 由框架接管,rt2 infershape 不处理)。

关联的Issue

#2135 https://gitcode.com/cann/ops-math/issues/2135

测试

Ascend 910B 真机验证全部通过:

  • op_host(infershape)UT:8/8
  • AICPU kernel UT:11/11(Conj 6 + ConjugateTranspose 5)
  • 打包 .run + 安装:SUCCESS;strings 自检两算子符号在 libmath_aicpu_kernels.so
  • 图模式 example:Conj、ConjugateTranspose 结果均正确,日志确认走 AICPU custom scheduler

文档更新

  • 新增 math/conj/README.mdconversion/conjugate_transpose/README.md
  • 更新 docs/zh/op_list.md(新增两算子条目)

类型标签

  • 新特性
likedislike
Pull Request已成功合入, 合并人@CANN-robot
(感谢 Ding_Jing 的贡献)
Ding_JingDing_Jing成员
7月3日 创建了 pull request,commit 7e5987cd
CANN-robotCANN-robot成员
7月3日 添加了label:cann-cla/yes
CANN-robot
CANN-robot成员
7月3日 评论:

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 陈娇, 宋恺 (2/2) 陈娇 (1/1)
*/*/op_graph/*_proto.h 汤磊, 宋恺 (2/2) 汤磊 (1/1)
repo-cann/ops-math 宋恺, 王林木 (2/2) 宋恺, 王林木 (2/1)

💡 Tip:

  • Committer can comment /approve or /lgtm
  • Commenting /approve implies both code review (lgtm) and intent to merge (approve)

CLA Signature Pass

Ding_Jing, thanks for your pull request. All authors of the commits have signed the CLA. 👍

likedislike
Ding_Jing
Ding_Jing成员
7月3日 评论:

compile

likedislike
Ding_JingDing_Jing成员
7月3日 update merge request[project id: 7649531, iid: 3778, commit_id: 3b6453b3115207194dba3a55812442ddfd73bcc4] virtual merging success
此处折叠了146条消息 查看更多
CANN-robotCANN-robot成员
7月10日 添加了label:approved
llimwang成员
7月10日 评论:

/approve

likedislike
CANN-robotCANN-robot成员
7月10日 添加了label:lgtm
CANN-robotCANN-robot成员
7月10日 关闭了关联的issue
CANN-robotCANN-robot成员
7月10日 合入了pull request