Merged
feat: pad_v2 新增 COMPLEX64/COMPLEX128 支持,circular_pad_grad 修改参数校验 #6513
tan_xinAdd on 22 days ago
feat: pad_v2 新增 COMPLEX64/COMPLEX128 支持,circular_pad_grad 修改参数校验 #6513
Merged
tan_xinAdd on 22 days ago
tan_xinCollaborators
22 days ago

描述

本次提交对 pad_v2 与 circular_pad_grad 算子进行功能增强与校验放宽。

改动原因

  • pad_v2 此前不支持 COMPLEX64/COMPLEX128,且 constant_values 被文档描述为必选,Tiling 侧存在冗余前置校验,限制适用场景
  • circular_pad_grad 的 shape 校验过严(padding 必须小于对应维度、gradOutput 前导维度必须与 self 一致),与实际允许场景不符

改动方法

  1. pad_v2 README 与 proto:x、constant_values、y 新增 COMPLEX64、COMPLEX128;constant_values 改为可选 Tensor(仅 constant 模式使用,缺省 0)
  2. pad_v2 Tiling:移除 paddings 空数据拦截、constant_values dtype 一致性与标量形状校验;example 增加 mode 属性
  3. circular_pad_grad op_api:移除 padding 小于对应维度、gradOutput 前导维度一致性校验,错误码修正为 ACLNN_ERR_INNER_NULLPTR
  4. 同步更新 tiling UT workspace 期望、golden.py、ST/UT 用例

关联的Issue

测试

文档更新

  • conversion/pad_v2/README.md
  • conversion/pad_v2/op_graph/pad_v2_proto.h
  • conversion/circular_pad_grad/docs/aclnnCircularPad2dBackward.md
  • conversion/circular_pad_grad/docs/aclnnCircularPad3dBackward.md

类型标签

likedislike
Pull Request has been successfully merged by CANN-robot.
(Thank you for tan_xin's contribution.)
Ttan_xinCollaborators
22 days ago Created pull request, commit 1ae526d7
atomgit-bot
atomgit-bot
22 days ago Comments:

变更摘要

本 PR 主要围绕 Tile 算子的形状推导(infershape)增加调试日志。改动集中在 math/tile/op_host/tile_infershape.cpp,在 TileInferShapeCommon 与 InferShape4Tile 中插入 OP_LOGI 日志,用于输出 multiples_len、multiples_data 各元素取值、维度计算的中间量以及最终输出维度;同时把原乘法表达式中的乘数提取为显式 int64_t 变量 multiplier 后再参与计算。

主要改动

  • 入口与成功日志:在 TileInferShapeCommon 函数入口与正常返回处新增 OP_LOGI,打印 multiples_len 及成功状态,便于确认函数调用与执行结果。
  • multiples_data 逐项打印:新增循环遍历 multiples_data,逐个输出其下标与取值。
  • 乘数计算重构与中间量日志:将原先内联的 multiples_data[i] 提取为 int64_t multiplier 并显式 static_cast,同时打印 in_shape_len、multiples_len、multiples_index、multiplier 及 multiples_data[multiples_index] 等信息,再通过 SetDim 写入并打印输出维度 out dim。
  • InferShape4Tile 入口日志:在 InferShape4Tile 入口新增 OP_LOGI,标记该分支进入 infershape 流程。

🧩 本次 Review 已使用自定义 Skill:

likedislike
Incorrect?
atomgit-bot
atomgit-bot
22 days ago Comments:

代码审查

✅ 未发现问题


🧩 本次 Review 已使用自定义 Skill:

likedislike
Incorrect?
CANN-robotCANN-robotCollaborators
22 days ago Added label: cann-cla/yes
Ttan_xinCollaborators
22 days ago Change title to "tile算子test", originally titled "tile算子加打印".
169 messages are collapsed here。 See more
CANN-robotCANN-robotCollaborators
13 days ago Added label: lgtm
tang-lei01Collaborators
13 days ago Comments:

/approve

likedislike
CANN-robotCANN-robotCollaborators
13 days ago Added label: approved
CANN-robotCANN-robotCollaborators
13 days ago Closed associated issue
CANN-robotCANN-robotCollaborators
13 days ago Merged pull request