已合并
fix: cat/cat.out回退手写实现路径 #40094
fix: cat/cat.out回退手写实现路径 #40094
已合并
zzhongmin创建于 7月4日
zzhongmin
zzhongmin成员
7月4日

【合入来源】

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

【修改方案】

不让 cat/cat.out 继续走 structured 自动代码生成链路,回到原本的手写实现路径

【资料变更】

不涉及

【接口变更】

不涉及

【功能验证】

BEVFusion网络显存正常

【CheckList】

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

likedislike
Pull Request已成功合入, 合并人@ascend-robot
(感谢 zzhongmin 的贡献)
zzhongminzzhongmin成员
7月4日 创建了 pull request,commit 3433176f
zzhongminzzhongmin成员
7月4日 关联了issue:[Bug]: 修复BEVFusion的显存劣化 700M问题
ascend-robotascend-robot成员
7月4日 添加了label:ascend-cla/yes
ascend-robot
ascend-robot成员
7月4日 评论:

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 ffmh, liangsongwei (2/2) liangsongwei (1/1)

💡 Tip:

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

CLA Signature Pass

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

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

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

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

评论 /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月4日 添加了label:ci-pipeline-running
ascend-robot
ascend-robot成员
7月4日 评论:

ascend docs pipeline is running...

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

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

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

变更摘要

此 PR 针对 catcat.out 两个算子,在 parse_backend_yaml 函数中强制绕过 structured 自动代码生成链路,使其回退到原本的手写实现路径,以修复 BEVFusion 网络中出现的显存异常问题。

主要改动

  • is_structured 计算逻辑修改:在 torchnpugen/gen_backend_stubs.pyparse_backend_yaml 函数中,对 is_structured 的赋值增加了对 op_name 的判断——当算子名称为 "cat.out" 时,直接将 is_structured 置为 False,从而阻止其进入 structured 代码生成分支。
  • has_structured_delegate 计算逻辑修改:同样在 parse_backend_yaml 函数中,对 has_structured_delegate 的赋值增加了对 op_name 的判断——当算子名称为 "cat" 时,直接将 has_structured_delegate 置为 False,使其不再走 structured delegate 路径。
likedislike
ascend-robotascend-robot成员
7月4日 删除了label:ci-pipeline-running
ascend-robotascend-robot成员
7月4日 添加了label:ci-pipeline-failed
ascend-robot
ascend-robot成员
7月4日 评论:
流水线 PR-pipeline_pytorch#41211 [ commitID:cd17014e ] 运行失败
>>>代码风格自动修复执行成功(无修复内容)
阶段 任务名 状态 详情
编译构建 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_X86_Part_01 🕚 >>>
UT_X86_Part_02 🕚 >>>
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]评论有效
  • compile、compile_inductor、compile_torchair : 运行流水线
  • retry : 重试流水线所有失败子任务
  • retry <任务名> : 仅重试指定失败子任务
  • stop : 停止流水线
likedislike
atomgit-bot
atomgit-bot
7月4日 评论:

代码审查

经过对 diff 及相关代码上下文的深入审查,我得出的结论是:此更改在逻辑上是正确的,且达到了预期目标。以下是我的分析摘要:

审查结果

变更分析

torchnpugen/gen_backend_stubs.py 第 450–451 行:

# 修改前:
is_structured = bool(op.get('structured', False))
has_structured_delegate = op.get('structured_delegate') is not None

# 修改后:
is_structured = bool(op.get('structured', False)) if op_name != "cat.out" else False
has_structured_delegate = op.get('structured_delegate') is not None if op_name != "cat" else False

变更意图:阻止 catcat.out 算子进入 structured 自动代码生成路径,使其回退到手写实现。这与 issue #2597(修复 BEVFusion 显存劣化 700M 问题)的目标一致。

逻辑正确性验证

  1. op_name 的来源(第 444 行):由 _get_backend_yaml_op_name(op) 提取,对于 yaml 中的 dict 条目,会取 func 字段 ( 之前的部分并 strip。对于 cat(Tensor[] tensors, int dim=0) -> Tensor 得到 "cat",对于 cat.out(...) 得到 "cat.out",与硬编码字符串完全匹配。

  2. 门控条件(第 452 行):if is_structured or has_structured_delegate: —— 只有当两者之一为 True 时才进入 structured 路径。

  3. cat.out 处理

    • is_structured 被强制设为 False
    • has_structured_delegate 正常检查(yaml 中 cat.out 不应有此字段)
    • 两者均为 False → 跳过 structured 路径 ✅
  4. cat 处理

    • is_structured 正常检查(yaml 中 cat 不应有 structured: True
    • has_structured_delegate 被强制设为 False
    • 两者均为 False → 跳过 structured 路径 ✅
  5. 下游影响:由于两者均不进入 structured 路径,check_structured_group_consistency(第 495 行)不会对 cat/cat.out 进行一致性检查,不会引发异常。

  6. Python 三元表达式优先级:解析正确,不存在结合性问题。

审查结论

未发现此 diff 引入的实际缺陷。该变更是针对特定算子(cat/cat.out)的定向回退方案,在当前的 yaml 配置下可正确生效。逻辑正确、语法无误、不影响其他算子的处理流程。


总结:审查了 1 个变更文件(torchnpugen/gen_backend_stubs.py),未发现问题。整体风险判断:。该变更是针对特定场景的精确修复,逻辑正确,无副作用。

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

likedislike
zzhongmin
zzhongmin成员
7月4日 评论:

compile

likedislike
ascend-robotascend-robot成员
7月4日 删除了label:ci-pipeline-failed
ascend-robotascend-robot成员
7月4日 添加了label:ci-pipeline-running
ascend-robot
ascend-robot成员
7月4日 评论:
流水线 PR-pipeline_pytorch#41218 [ commitID:cd17014e ] 运行中
>>>代码风格自动修复执行成功(无修复内容)
阶段 任务名 状态 详情
编译构建 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_X86_Part_01 🕚 >>>
UT_X86_Part_02 🕚 >>>
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]评论有效
  • compile、compile_inductor、compile_torchair : 运行流水线
  • retry : 重试流水线所有失败子任务
  • retry <任务名> : 仅重试指定失败子任务
  • stop : 停止流水线
likedislike
ascend-robot
ascend-robot成员
7月4日 评论:

ascend docs pipeline is running...

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

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

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

retry

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

重试流水线失败:PR-pipeline_pytorch 仅失败状态流水线支持重试操作。

likedislike
ascend-robotascend-robot成员
7月4日 删除了label:ci-pipeline-running
ascend-robotascend-robot成员
7月4日 添加了label:ci-pipeline-passed
ascend-robot
ascend-robot成员
7月4日 评论:
流水线 PR-pipeline_pytorch#41218 (重试第1次) [ commitID:cd17014e ] 已完成
>>>代码风格自动修复执行成功(无修复内容)
阶段 任务名 状态 详情
编译构建 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_X86_Part_01 🛑 >>>
UT_X86_Part_02 🛑 >>>
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]评论有效
  • compile、compile_inductor、compile_torchair : 运行流水线
  • retry : 重试流水线所有失败子任务
  • retry <任务名> : 仅重试指定失败子任务
  • stop : 停止流水线
likedislike
梁松伟
梁松伟成员
7月4日 评论:

/approve

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

/lgtm

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