已合并
add_npu_backend_init_log #39403
cuiduo创建于 6月26日
add_npu_backend_init_log #39403
已合并
cuiduo创建于 6月26日
cuiduo
cuiduo成员
6月26日

【合入来源】

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

【修改方案】

修复dvm后端stream报错的问题,增加使用后端提示信息

【资料变更】

不涉及

【接口变更】

不涉及

【功能验证】

UT已通过

【CheckList】

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

likedislike
Pull Request已成功合入, 合并人@ascend-robot
(感谢 cuiduo 的贡献)
cuiduocuiduo成员
6月26日 创建了 pull request,commit 341972a2
ascend-robotascend-robot成员
6月26日 添加了label:ascend-cla/yes
ascend-robot
ascend-robot成员
6月26日 评论:

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
test zichun_ye, suhaibo (2/2) zichun_ye (1/1)
torch_npu/utils zichun_ye, suhaibo (2/2) zichun_ye (1/1)

💡 Tip:

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

CLA Signature Pass

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

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

Linking Issue Notice

@cuiduo , 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成员
6月26日 评论:

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

Protected Branch Version Release
master
v2.7.1-26.1.0
v2.12.0-26.1.0
v2.11.0-26.1.0
v2.10.0-26.1.0
v2.9.0-26.1.0
v2.12.0
v2.11.0
v2.10.0
v2.9.0
v2.7.1
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成员
6月26日 添加了label:ci-pipeline-running
ascend-robot
ascend-robot成员
6月26日 评论:

Ascend docs pipeline is running...

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

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

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

变更摘要

此 PR 主要为 NPU 后端初始化过程添加了日志记录能力。在 torch_npu/utils/_dynamo.py 中,patch_inductor_wrapper() 内部的 new_init 方法在解析到 MLIR 或 DVM 后端时,通过 log.info(...) 输出对应的后端运行日志,便于问题定位和运维监控。同时,测试文件 test/_inductor/test_add_sum.py 进行了一处微小清理(删除一行)。

主要改动

  • NPU 后端初始化日志: 在 patch_inductor_wrapper()new_init 方法中,当后端解析为 "mlir""dvm" 时,分别添加 log.info("Running MLIR backend")log.info("Running dvm backend") 日志输出,帮助开发者追踪实际运行的后端类型。
  • 测试文件清理: test/_inductor/test_add_sum.py 删除了一个无用行(-1 行),属于代码清理性质。
likedislike
atomgit-bot
atomgit-bot
6月26日 评论:

代码审查


审查总结

优先级 数量 说明
P0 1 log 变量未定义,运行时必定崩溃
P1 1 dvm 后端丢失流恢复调用,行为回归
P2 0 -
P3 0 -

已审查文件:

  • test/_inductor/test_add_sum.py — 无问题(移除的 import torch_npu 已在父类模块 testutils.py 中导入,冗余删除安全)
  • torch_npu/utils/_dynamo.py — 发现 2 个问题(1 个 P0 + 1 个 P1)

整体风险评估: 高风险。P0 问题将导致 mlir/dvm 后端初始化时直接崩溃,P1 问题可能导致 dvm 后端运行时出现流未恢复的异常行为。建议在合入前修复这两个问题。

类型 数量
🔴 阻塞 2
🟡 建议 0

⛔ 需要修改

likedislike
atomgit-bot
atomgit-bot6月26日进行代码检视1
torch_npu/utils/_dynamo.py
@@ -230,2 +230,3 @@
230- if backend == "mlir" or backend == "dvm":
230+ if backend=="mlir":
231231 with _NpuBackendScope(backend):
232+ log.info("Running MLIR backend")
atomgit-bot
atomgit-bot6月26日评论:

🔴 Critical

文件 torch_npu/utils/_dynamo.py 中未定义 log 变量——没有 import logging,也没有 log = logging.getLogger(...) 或从其他模块导入。第232行和第237行的 log.info(...) 调用将在运行时触发 NameError: name 'log' is not defined,导致整个 new_init 函数崩溃,inductor wrapper 初始化失败。

建议:在文件顶部添加 import logging,并在模块级别创建 logger:log = logging.getLogger(__name__),或者使用 torch._logging.getArtifactLogger(与 _graph_tree.py 中的模式一致)。

  当前建议代码无改动
likedislike
atomgit-bot
atomgit-bot6月26日进行代码检视1
torch_npu/utils/_dynamo.py
@@ -233,1 +234,4 @@
233234 torch_npu._C._recovery_all_npu_stream(device_id)
235+ if backend=="dvm":
236+ with _NpuBackendScope(backend):
237+ log.info("Running dvm backend")
atomgit-bot
atomgit-bot6月26日评论:

🟠 High Priority

原始代码中 mlirdvm 后端共享同一个代码块,均执行了 device_id = torch_npu.npu.current_device()torch_npu._C._recovery_all_npu_stream(device_id) 进行流恢复。变更将 or 条件拆分为两个独立 if 块后,dvm 分支(第235-237行)仅保留了 _NpuBackendScope 上下文进入和新增的日志输出,丢失了流恢复调用。而 mlir 分支(第230-234行)仍保留了完整的流恢复逻辑。PR 标题为"add_npu_backend_init_log",意图仅为添加日志,表明这是拆分条件时意外遗漏。若 dvm 后端依赖流恢复,将导致运行时行为异常(如流未正确恢复导致的死锁或数据竞争)。

建议:在 dvm 分支中补回 device_id = torch_npu.npu.current_device()torch_npu._C._recovery_all_npu_stream(device_id) 调用,与 mlir 分支保持一致;或者如果确认 dvm 确实不需要流恢复,请在代码注释中说明原因。

likedislike
cuiduocuiduo成员
6月26日 update merge request[project id: 7404318, iid: 39403, commit_id: 3b8a25ed0bb8d27bf77f1b6984c0f326aabb1540] virtual merging success
cuiduocuiduo成员
6月26日 强制推送  1 个提交:3399816f-add_npu_backend_init_log
cuiduocuiduo成员
6月26日 update merge request[project id: 7404318, iid: 39403, commit_id: a81fc061a0facd10a4926bb557e48a1b3d8d5b8f] virtual merging success
ascend-robotascend-robot成员
6月26日 删除了label:ci-pipeline-running
ascend-robot
ascend-robot成员
6月26日 评论:
流水线 PR-pipeline_pytorch#39390 [ commitID:5763d86a ] 已终止运行
>>>代码风格自动修复执行成功(无修复内容)
阶段 任务名 状态 详情
编译构建 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]评论有效
  • compile、compile_inductor、compile_torchair : 运行流水线
  • retry : 重试流水线所有失败子任务
  • retry <任务名> : 仅重试指定失败子任务
  • stop : 停止流水线
likedislike
cuiduo
cuiduo成员
6月26日 评论:

compile

likedislike
ascend-robot
ascend-robot成员
6月26日 评论:

Ascend docs pipeline is running...

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

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

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

Ascend docs pipeline is running...

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

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

likedislike
ascend-robotascend-robot成员
6月26日 删除了label:docs-ci-pipeline-running
ascend-robotascend-robot成员
6月26日 添加了label:docs-ci-pipeline-success
ascend-robotascend-robot成员
6月26日 删除了label:ci-pipeline-running
ascend-robotascend-robot成员
6月26日 添加了label:ci-pipeline-passed
ascend-robot
ascend-robot成员
6月26日 评论:
流水线 PR-pipeline_pytorch#39400 [ commitID:3399816f ] 已完成
>>>代码风格自动修复执行成功(无修复内容)
阶段 任务名 状态 详情
编译构建 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]评论有效
  • compile、compile_inductor、compile_torchair : 运行流水线
  • retry : 重试流水线所有失败子任务
  • retry <任务名> : 仅重试指定失败子任务
  • stop : 停止流水线
likedislike
cuiduocuiduo成员
6月27日 修改了pull request 的描述
cuiduocuiduo成员
6月27日 删除了label:needs-issue
cuiduocuiduo成员
6月27日 关联了issue:[Bug]: 修复inductor图模式bug
zichun_ye
zichun_ye成员
6月27日 评论:

/approve

likedislike
ascend-robotascend-robot成员
6月27日 添加了label:approved
cuiduocuiduo成员
6月27日 解决了最后一个问题
suhaibo成员
6月27日 评论:

/lgtm

likedislike
ascend-robotascend-robot成员
6月27日 添加了label:lgtm
ascend-robotascend-robot成员
6月27日 合入了pull request
ascend-robot
ascend-robot成员
6月27日 评论:

Pull Request 已合并或已关闭。

If you want to solve this problem, you can click here to do it in the FAQs.

likedislike
ascend-robot
ascend-robot成员
6月27日 评论:
流水线 pytorch_gitcode_PR_multiVersion#11421 [ commitID:3399816f ] 已完成
likedislike