已合并
fix: Fall back to the aclop path for reshaped Nx1 column vectors #42471
fix: Fall back to the aclop path for reshaped Nx1 column vectors #42471
已合并
21xiu创建于 7月23日
21xiu
21xiu成员
7月23日

【合入来源】

如有社区issue,请关联issue链接
请勿携带内部流程信息(需求链接、问题单、内部issue等)

【修改方案】

shape为[1,n]的tensor做transpose()后再做ND转NZ当前有精度问题,回退至aclop

【资料变更】

不涉及

【接口变更】

不涉及

【功能验证】

新增用例pytest test/custom_ops/test_npu_format_cast.py -k test_2d_view_nx1_nd_to_nz_format_id
修改前:
ScreenShot_20260723160440.JPG
修改后:
ScreenShot_20260723160705.JPG

【CheckList】

PR提交人对以下CheckList自检项进行全量自检,自检通过或不涉及,均修改 [ ] 为 [x]

likedislike
Pull Request已成功合入, 合并人@ascend-robot
(感谢 21xiu 的贡献)
21xiu21xiu成员
7月23日 创建了 pull request,commit 33ff5740
atomgit-bot
atomgit-bot
7月23日 评论:

变更摘要

此 PR 在 FormatCastKernelNpu.cppMaybeUseAclnnNpuFormatCast 函数中新增了一个防护判断:当对形状与存储形状不一致的 2D 连续张量执行 ND → FRACTAL_NZ 格式转换时(例如 [1,N]transpose() 后变为 [N,1] 的列向量),回退到非 aclnn 路径,以规避下游算子的精度问题。

主要改动

  • 新增形状与存储形状一致性检查:在 MaybeUseAclnnNpuFormatCast 中,当源格式为 ACL_FORMAT_ND、目标格式为 ACL_FORMAT_FRACTAL_NZ、张量为 2D 连续且最后一维大小为 1 时,比对 src.sizes()src_desc.base_sizes_,若两者不一致则判定为 view 操作(如 transpose),返回 false 回退到非 aclnn 路径
  • 保护范围限定为 N×1 列向量场景:条件中通过 src.sizes()[1] == 1 将防护范围限定为 reshape 后的 N×1 列向量,避免对其他形状的 ND→NZ 转换造成不必要的影响
  • 无删除、仅新增:此改动为纯增量修改,共 12 行新增代码,不影响现有逻辑路径
likedislike
atomgit-bot
atomgit-bot
7月23日 评论:

代码审查

经过对整个变更的深入分析,包括上下文代码、调用链、数据结构定义以及边界条件的完整审查,我未发现该 diff 引入的任何逻辑错误、安全风险、可靠性问题或性能退化。

变更内容是在 MaybeUseAclnnNpuFormatCast 函数中新增一个守卫条件:当对 shape 为 [N, 1] 的 2D 连续 tensor 做 ND→FRACTAL_NZ 格式转换时,若其 storage 的 base_sizes_ 与当前 sizes() 不一致(即为 view/transpose 产生),则回退到非 aclnn 路径,以避免精度问题。

审查结论:未发现需要报告的问题。

各文件审查结果

  • torch_npu/csrc/aten/common/FormatCastKernelNpu.cpp:+12 行,无问题。

Closing Summary

  • 发现数量(按严重性):P0: 0, P1: 0, P2: 0, P3: 0
  • 整体风险判断:低风险。该变更是针对性的防御性回退,逻辑正确,条件检查完整(包含 size 边界保护),与现有代码风格一致,调用方对回退路径的处理亦正确。

⚠️ 已识别出整体风险,但无法提取行内评论,请参考整体评估。

likedislike
ascend-robotascend-robot成员
7月23日 添加了label:ascend-cla/yes
ascend-robotascend-robot成员
7月23日 添加了label:stat/needs-squash
ascend-robot
ascend-robot成员
7月23日 评论:

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
repo-Ascend/pytorch 楚浩田, chengpeng25 (2/2) 楚浩田 (1/1)
test 楚浩田, chengpeng25 (2/2) 楚浩田 (1/1)

💡 Tip:

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

CLA Signature Pass

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

likedislike
ascend-robotascend-robot成员
7月23日 添加了label:needs-issue
ascend-robot
ascend-robot成员
7月23日 评论:

Linking Issue Notice

@xiu_21 , the pull request must be linked to at least one issue.
If an issue has already been linked, but the needs-issue label remains, you can remove the label by commenting /check-issue .

likedislike
ascend-robot
ascend-robot成员
7月23日 评论:

当前仓库存在以下 保护分支

Protected Branch Version Release
master
v2.9.0
v2.12.0
v2.11.0
v2.7.1
v2.10.0
v2.7.1-26.1.0
v2.9.0-26.1.0
v2.12.0-26.1.0
v2.10.0-26.1.0
v2.11.0-26.1.0
ci-test

评论 /sync <branch1> <branch2> ... 可将当前 PR 修改同步到其它分支(创建同步 PR):
a) 如果当前 PR 是 Open 状态,同步操作将延迟到 PR 被合并时执行
b) 如果当前 PR 已经 Merged,将立即执行同步操作

注意:

  1. /sync 命令可以指定同步到多个分支,仅最后一个 /sync 命令生效
  2. 如果创建的同步 PR 不正确,可通过向同步 PR 的源分支提交轻量级 PR 完善,或使用 /close 命令关闭
likedislike
ascend-robotascend-robot成员
7月23日 添加了label:ci-pipeline-running
ascend-robot
ascend-robot成员
7月23日 评论:

ascend docs pipeline is running...

likedislike
ascend-robotascend-robot成员
7月23日 添加了label:docs-ci-pipeline-running
21xiu21xiu成员
7月23日 关联了issue:[Bug]: shape为[1,n]的tensor做transpose()后再做ND转NZ当前有精度问题
ascend-robotascend-robot成员
7月23日 删除了label:needs-issue
ascend-robot
ascend-robot成员
7月23日 评论:

✅ 跳过 docs ci 检查,没有需要检查的文档文件

likedislike
ascend-robotascend-robot成员
7月23日 删除了label:docs-ci-pipeline-running
ascend-robotascend-robot成员
7月23日 添加了label:docs-ci-pipeline-success
21xiu21xiu成员
7月23日 修改标题为 “fix: Fall back to the aclop path for reshaped Nx1 column vectors”,原标题为“fix: Fall back to the non-aclnn path for reshaped Nx1 column vectors”
21xiu21xiu成员
7月23日 修改了pull request 的描述
ascend-robotascend-robot成员
7月23日 删除了label:ci-pipeline-running
ascend-robotascend-robot成员
7月23日 添加了label:ci-pipeline-passed
ascend-robot
ascend-robot成员
7月23日 评论:
流水线 PR-pipeline_pytorch#48479 [ commitID:ea5ee6ea ] 已完成
>>>代码风格自动修复执行成功(无修复内容)
阶段 任务名 状态 详情
编译构建 Build_X86 >>>
Build_ARM >>>
Build_LibTorch_x86 >>>
Build_LibTorch_ARM >>>
Build_X86_torchair 🛑 >>>
Build_ARM_torchair 🛑 >>>
patch_test 🛑 >>>
恶意代码检查 Antipoison >>>
编码安全与规范检查 CodeCheck >>>
check_error >>>
CodeCheck_lintrunner >>>
开源片段检查 SCA >>>
开发者测试 UT_ARM_A3_Part_01 🛑 >>>
UT_ARM_A3_Part_02 🛑 >>>
UT_ARM_A2_Part_01 >>>
UT_ARM_A2_Part_02 >>>
UT_ARM_A2_Part_03 >>>
UT_inductor_Part_01 🛑 >>>
UT_inductor_Part_02 🛑 >>>
UT_inductor_Part_03 🛑 >>>
UT_inductor_Part_04 🛑 >>>
UT_DIST_ARM_Part_01 🛑 >>>
UT_DIST_ARM_Part_02 🛑 >>>
UT_DIST_ARM_Part_03 🛑 >>>
UT_DIST_ARM_Part_04 🛑 >>>
UT_ARM_A2_Select_Part_01 >>>
UT_ARM_A2_Select_Part_02 >>>
流水线 PR-pipeline_pytorch >>>
此流水线已支持下列评论快捷指令,仅PR创建者和白名单成员[wujinyuan1, huangjingwei, liangsongwei, yashi999, culechan, Dring, wuyouqi1, L1919_snow, qq_52711437, WhiteNight12, nomiz, xiu_21, ffmh, wanglijun55, hss-shuai, husichao, smallsilly, lanshaozuishuai, jimmyisme1, lzy0920232, alpha-junh, Sunshine_Youngster, wei_zhuoyi, zhangyihuiben, zyw-hw, zzzkeke, rmch, yangch0324, LucciC, AACAES, renyujin, wjlflyer, senzhen-town, pengjingyou, qsc97, limuan, yule100, xiaoqi-zhou, kuhn7, chenxingying, hanye02, zichun_ye, anyrenwei, kkjocker, wangzili121, Lu_G, yvjc, puddingfjz, HandsoemLemon, bigprestigee1, huawuyi, zhenyu10, dairenjie, du-jin-hang, zou-jieyu, adelaideliu, TrHan, wanlinan, Windwindzzz, pengqihw, kisnwang, yuheng_wang, honghao_wang, jizewei, zhangguoguang, sunyu-xuan, chenrayray, hbhu_bin, liujunzhu, c_34, LiNuoh, maoyuanpeng1, zzhongmin, zhaoyu65, bellatan, jiabaolin, zhuofanshen, wencaiwen, lu_zhuge, caoshuyang, molly12, lyx324521, LQ1206, gitcode-chenjiao, cai-weiwei1989, CHDong, ogqin, yuanlipingGit, xuqinglin1, lqz2, zouwei1, chaoluoa, paradox325, jackzhang1116, yaoyao, akh, yujiacheng, dengjie0116, Hubert11111, Shine_Ws, wslhj555, longqiand, OYtao666, JiaqingQiang, luyyyy, Kingbelial, zhanghaiyu0101, wenxp1018, yanliu-luoluo, ksun_sekiro, liyong328, wgzheng, tangky, vivi_is_coding, aoiaoisola, weixin_44494597, wangmengmengwang65667, hid57809721, qq_35468730, comeonup, C547032, gcw_m5OQChA4, yao_yao_ling_xian, cnnbwcy, szqfes_12, cora_19, cann_lilin, can, shawnylee233, fanglanyue0916, hhz0, LiNuohang, taohuoquan]评论有效
  • compile、compile_inductor、compile_torchair : 运行流水线
  • retry : 重试流水线所有失败子任务
  • retry <任务名> : 仅重试指定失败子任务
  • stop : 停止流水线
likedislike
21xiu21xiu成员
7月23日 update merge request[project id: 7404318, iid: 42471, commit_id: 10b787671f22d5ade2146b9e907bbe73b555b0d8] virtual merging success
21xiu21xiu成员
7月23日 推送  1 个提交:b606c770-fix: Fall back to the non-aclnn path for reshaped Nx1 column vectors
21xiu21xiu成员
7月23日 update merge request[project id: 7404318, iid: 42471, commit_id: 17640de334fe3a0006e6a7b84cca0b96e751ca37] virtual merging success
ascend-robotascend-robot成员
7月23日 删除了label:ci-pipeline-passed
21xiu
21xiu成员
7月23日 评论:

compile

likedislike
ascend-robotascend-robot成员
7月23日 添加了label:ci-pipeline-running
ascend-robot
ascend-robot成员
7月23日 评论:

ascend docs pipeline is running...

likedislike
ascend-robotascend-robot成员
7月23日 删除了label:docs-ci-pipeline-success
ascend-robotascend-robot成员
7月23日 添加了label:docs-ci-pipeline-running
ascend-robot
ascend-robot成员
7月23日 评论:

ascend docs pipeline is running...

likedislike
ascend-robot
ascend-robot成员
7月23日 评论:

✅ 跳过 docs ci 检查,没有需要检查的文档文件

likedislike
ascend-robotascend-robot成员
7月23日 删除了label:docs-ci-pipeline-running
ascend-robotascend-robot成员
7月23日 添加了label:docs-ci-pipeline-success
ascend-robot
ascend-robot成员
7月23日 评论:

✅ 跳过 docs ci 检查,没有需要检查的文档文件

likedislike
21xiu21xiu成员
7月23日 修改了pull request 的描述
ascend-robotascend-robot成员
7月23日 删除了label:ci-pipeline-running
ascend-robotascend-robot成员
7月23日 添加了label:ci-pipeline-passed
ascend-robot
ascend-robot成员
7月23日 评论:
流水线 PR-pipeline_pytorch#48627 [ commitID:b606c770 ] 已完成
>>>代码风格自动修复执行成功(无修复内容)
阶段 任务名 状态 详情
编译构建 Build_X86 >>>
Build_ARM >>>
Build_LibTorch_x86 >>>
Build_LibTorch_ARM >>>
Build_X86_torchair 🛑 >>>
Build_ARM_torchair 🛑 >>>
patch_test 🛑 >>>
恶意代码检查 Antipoison >>>
编码安全与规范检查 CodeCheck >>>
check_error >>>
CodeCheck_lintrunner >>>
开源片段检查 SCA >>>
开发者测试 UT_ARM_A3_Part_01 🛑 >>>
UT_ARM_A3_Part_02 🛑 >>>
UT_ARM_A2_Part_01 >>>
UT_ARM_A2_Part_02 >>>
UT_ARM_A2_Part_03 >>>
UT_inductor_Part_01 🛑 >>>
UT_inductor_Part_02 🛑 >>>
UT_inductor_Part_03 🛑 >>>
UT_inductor_Part_04 🛑 >>>
UT_DIST_ARM_Part_01 🛑 >>>
UT_DIST_ARM_Part_02 🛑 >>>
UT_DIST_ARM_Part_03 🛑 >>>
UT_DIST_ARM_Part_04 🛑 >>>
UT_ARM_A2_Select_Part_01 >>>
UT_ARM_A2_Select_Part_02 >>>
流水线 PR-pipeline_pytorch >>>
此流水线已支持下列评论快捷指令,仅PR创建者和白名单成员[wujinyuan1, huangjingwei, liangsongwei, yashi999, culechan, Dring, wuyouqi1, L1919_snow, qq_52711437, WhiteNight12, nomiz, xiu_21, ffmh, wanglijun55, hss-shuai, husichao, smallsilly, lanshaozuishuai, jimmyisme1, lzy0920232, alpha-junh, Sunshine_Youngster, wei_zhuoyi, zhangyihuiben, zyw-hw, zzzkeke, rmch, yangch0324, LucciC, AACAES, renyujin, wjlflyer, senzhen-town, pengjingyou, qsc97, limuan, yule100, xiaoqi-zhou, kuhn7, chenxingying, hanye02, zichun_ye, anyrenwei, kkjocker, wangzili121, Lu_G, yvjc, puddingfjz, HandsoemLemon, bigprestigee1, huawuyi, zhenyu10, dairenjie, du-jin-hang, zou-jieyu, adelaideliu, TrHan, wanlinan, Windwindzzz, pengqihw, kisnwang, yuheng_wang, honghao_wang, jizewei, zhangguoguang, sunyu-xuan, chenrayray, hbhu_bin, liujunzhu, c_34, LiNuoh, maoyuanpeng1, zzhongmin, zhaoyu65, bellatan, jiabaolin, zhuofanshen, wencaiwen, lu_zhuge, caoshuyang, molly12, lyx324521, LQ1206, gitcode-chenjiao, cai-weiwei1989, CHDong, ogqin, yuanlipingGit, xuqinglin1, lqz2, zouwei1, chaoluoa, paradox325, jackzhang1116, yaoyao, akh, yujiacheng, dengjie0116, Hubert11111, Shine_Ws, wslhj555, longqiand, OYtao666, JiaqingQiang, luyyyy, Kingbelial, zhanghaiyu0101, wenxp1018, yanliu-luoluo, ksun_sekiro, liyong328, wgzheng, tangky, vivi_is_coding, aoiaoisola, weixin_44494597, wangmengmengwang65667, hid57809721, qq_35468730, comeonup, C547032, gcw_m5OQChA4, yao_yao_ling_xian, cnnbwcy, szqfes_12, cora_19, cann_lilin, can, shawnylee233, fanglanyue0916, hhz0, LiNuohang, taohuoquan]评论有效
  • compile、compile_inductor、compile_torchair : 运行流水线
  • retry : 重试流水线所有失败子任务
  • retry <任务名> : 仅重试指定失败子任务
  • stop : 停止流水线
likedislike
htchu成员
7月23日 评论:

/approve

likedislike
ascend-robotascend-robot成员
7月23日 添加了label:approved
chengpeng25成员
7月23日 评论:

/lgtm

likedislike
ascend-robotascend-robot成员
7月23日 添加了label:lgtm
ascend-robotascend-robot成员
7月23日 合入了pull request
ascend-robot
ascend-robot成员
7月23日 评论:
流水线 pytorch_gitcode_PR_multiVersion#13035 [ commitID:b606c770 ] 已完成
likedislike