已合并
[fix]mlir_enable #31966
cuiduo创建于 3月18日
[fix]mlir_enable #31966
已合并
cuiduo创建于 3月18日
cuiduo
cuiduo成员
3月18日

【合入来源】

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

【修改方案】

请修改MLIR使能方式,新增MLIR测试用例进行维护

【资料变更】

不涉及

【接口变更】

不涉及”

【功能验证】

已包含UT

【CheckList】

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

likedislike
Pull Request已成功合入, 合并人@ascend-robot
(感谢 cuiduo 的贡献)
cuiduocuiduo成员
3月18日 创建了 pull request,commit 9084d268
ascend-robot
ascend-robot成员
3月18日 评论:

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 adrian07110, zichun_ye (2/2) zichun_ye, adrian07110 (2/1)
torch_npu/_inductor zichun_ye, adrian07110 (2/2) zichun_ye, adrian07110 (2/1)
torch_npu/utils zichun_ye, adrian07110 (2/2) adrian07110, zichun_ye (2/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成员
3月18日 添加了label:ascend-cla/yes
cuiduo
cuiduo成员
3月18日 评论:

compile

likedislike
AtlasAccount
AtlasAccount成员
3月18日 评论:

问题/功能描述

本PR为NPU后端新增了对MLIR编译路径的支持,解决了torch.compile无法通过配置选项或环境变量启用MLIR后端进行算子融合编译的问题,为用户提供了更灵活的编译后端选择。

修改方案描述

核心修改是在torch_npu/utils/_dynamo.pypatch_inductor_wrapper函数中,当检测到npu_backend配置为”mlir”时,设置相应的环境变量并动态导入torch_mlir模块,以启用MLIR编译流程。同时,新增了单元测试文件test/_inductor/test_mlir_enable.py,验证通过options参数或全局配置启用MLIR后端的功能正确性。

likedislike
ascend-robotascend-robot成员
3月18日 添加了label:ci-pipeline-running
AtlasAccount
AtlasAccount成员
3月18日 评论:

/torch_npu/utils/_dynamo.py

likedislike
ascend-robotascend-robot成员
3月18日 删除了label:ci-pipeline-running
ascend-robotascend-robot成员
3月18日 添加了label:ci-pipeline-failed
ascend-robot
ascend-robot成员
3月18日 评论:
流水线 PR-pipeline_pytorch#11275 运行失败
阶段 任务名 状态 详情
编译构建 Build_X86 >>>
Build_ARM >>>
Build_LibTorch_x86 >>>
Build_X86_torchair 🛑 >>>
Build_ARM_torchair 🛑 >>>
Build_ARM_inductor >>>
恶意代码检查 Antipoison >>>
编码安全与规范检查 CodeCheck >>>
check_error >>>
开源片段检查 SCA >>>
开发者测试 UT_X86_Part_01 🛑 >>>
UT_X86_Part_02 🛑 >>>
UT_ARM_A3_Part_01 🛑 >>>
UT_ARM_A3_Part_02 🛑 >>>
UT_DIST_X86_Part_01 🛑 >>>
UT_DIST_X86_Part_02 🛑 >>>
UT_DIST_X86_Part_03 🛑 >>>
UT_DIST_X86_Part_04 🛑 >>>
UT_inductor_Part_01 >>>
UT_inductor_Part_02 >>>
UT_inductor_Part_03 >>>
UT_inductor_Part_04 >>>
UT_ARM_A2_Part_01 >>>
UT_ARM_A2_Part_02 >>>
流水线 PR-pipeline_pytorch >>>
likedislike
cuiduo
cuiduo成员
3月18日 评论:

compile

likedislike
ascend-robotascend-robot成员
3月18日 删除了label:ci-pipeline-failed
ascend-robotascend-robot成员
3月18日 添加了label:ci-pipeline-running
AtlasAccount
AtlasAccount成员
3月18日 评论:

/torch_npu/utils/_dynamo.py

likedislike
cuiduocuiduo成员
3月18日 update merge request[project id: 7404318, iid: 31966, commit_id: c7c172ae8bad60e1ddcc6ebd11eed7ee38acbb4e] virtual merging success
此处折叠了5条事件消息 查看更多
ascend-robotascend-robot成员
3月18日 添加了label:ci-pipeline-running
cuiduo
cuiduo成员
3月18日 评论:

compile

likedislike
cuiduo
cuiduo成员
3月18日 评论:

compile

likedislike
AtlasAccount
AtlasAccount成员
3月18日 评论:

/torch_npu/utils/_dynamo.py

likedislike
ascend-robotascend-robot成员
3月18日 删除了label:ci-pipeline-running
ascend-robot
ascend-robot成员
3月18日 评论:
流水线 PR-pipeline_pytorch#11296 已终止运行
阶段 任务名 状态 详情
编译构建 Build_X86 >>>
Build_ARM >>>
Build_LibTorch_x86 >>>
Build_X86_torchair 🛑 >>>
Build_ARM_torchair 🛑 >>>
Build_ARM_inductor >>>
恶意代码检查 Antipoison >>>
编码安全与规范检查 CodeCheck >>>
check_error >>>
开源片段检查 SCA >>>
开发者测试 UT_X86_Part_01 🛑 >>>
UT_X86_Part_02 🛑 >>>
UT_ARM_A3_Part_01 🛑 >>>
UT_ARM_A3_Part_02 🛑 >>>
UT_DIST_X86_Part_01 🛑 >>>
UT_DIST_X86_Part_02 🛑 >>>
UT_DIST_X86_Part_03 🛑 >>>
UT_DIST_X86_Part_04 🛑 >>>
UT_inductor_Part_01 🟨 >>>
UT_inductor_Part_02 >>>
UT_inductor_Part_03 >>>
UT_inductor_Part_04 >>>
UT_ARM_A2_Part_01 >>>
UT_ARM_A2_Part_02 >>>
流水线 PR-pipeline_pytorch 🟨 >>>
likedislike
ascend-robotascend-robot成员
3月18日 添加了label:ci-pipeline-running
cuiduocuiduo成员
3月18日 修改了pull request 的描述
ascend-robotascend-robot成员
3月18日 删除了label:ci-pipeline-running
ascend-robotascend-robot成员
3月18日 添加了label:ci-pipeline-failed
ascend-robot
ascend-robot成员
3月18日 评论:
流水线 PR-pipeline_pytorch#11326 运行失败
阶段 任务名 状态 详情
编译构建 Build_X86 >>>
Build_ARM >>>
Build_LibTorch_x86 >>>
Build_X86_torchair 🛑 >>>
Build_ARM_torchair 🛑 >>>
Build_ARM_inductor >>>
恶意代码检查 Antipoison >>>
编码安全与规范检查 CodeCheck >>>
check_error >>>
开源片段检查 SCA >>>
开发者测试 UT_X86_Part_01 🛑 >>>
UT_X86_Part_02 🛑 >>>
UT_ARM_A3_Part_01 🛑 >>>
UT_ARM_A3_Part_02 🛑 >>>
UT_DIST_X86_Part_01 🛑 >>>
UT_DIST_X86_Part_02 🛑 >>>
UT_DIST_X86_Part_03 🛑 >>>
UT_DIST_X86_Part_04 🛑 >>>
UT_inductor_Part_01 >>>
UT_inductor_Part_02 >>>
UT_inductor_Part_03 >>>
UT_inductor_Part_04 >>>
UT_ARM_A2_Part_01 >>>
UT_ARM_A2_Part_02 >>>
流水线 PR-pipeline_pytorch >>>
likedislike
cuiduo
cuiduo成员
3月18日 评论:

compile

likedislike
ascend-robotascend-robot成员
3月18日 删除了label:ci-pipeline-failed
ascend-robotascend-robot成员
3月18日 添加了label:ci-pipeline-running
AtlasAccount
AtlasAccount成员
3月18日 评论:

/torch_npu/utils/_dynamo.py

likedislike
ascend-robotascend-robot成员
3月19日 删除了label:ci-pipeline-running
ascend-robotascend-robot成员
3月19日 添加了label:ci-pipeline-passed
ascend-robot
ascend-robot成员
3月19日 评论:
流水线 PR-pipeline_pytorch#11400 已完成
阶段 任务名 状态 详情
编译构建 Build_X86 >>>
Build_ARM >>>
Build_LibTorch_x86 >>>
Build_X86_torchair 🛑 >>>
Build_ARM_torchair 🛑 >>>
Build_ARM_inductor >>>
恶意代码检查 Antipoison >>>
编码安全与规范检查 CodeCheck >>>
check_error >>>
开源片段检查 SCA >>>
开发者测试 UT_X86_Part_01 🛑 >>>
UT_X86_Part_02 🛑 >>>
UT_ARM_A3_Part_01 🛑 >>>
UT_ARM_A3_Part_02 🛑 >>>
UT_DIST_X86_Part_01 🛑 >>>
UT_DIST_X86_Part_02 🛑 >>>
UT_DIST_X86_Part_03 🛑 >>>
UT_DIST_X86_Part_04 🛑 >>>
UT_inductor_Part_01 >>>
UT_inductor_Part_02 >>>
UT_inductor_Part_03 >>>
UT_inductor_Part_04 >>>
UT_ARM_A2_Part_01 >>>
UT_ARM_A2_Part_02 >>>
流水线 PR-pipeline_pytorch >>>
likedislike
zichun_ye
zichun_ye成员
3月19日 评论:

/approve

likedislike
ascend-robotascend-robot成员
3月19日 添加了label:approved
ZhihaoLi
ZhihaoLi成员
3月19日 评论:

/approve

likedislike
ascend-robotascend-robot成员
3月19日 添加了label:lgtm
ascend-robot
ascend-robot成员
3月19日 评论:

Review Guide

This pull-request passes review.
Committers who wrote a comment of /approve are: zichun_ye, adrian07110.
Reviewers who wrote a comment of /lgtm are: zichun_ye, adrian07110.

likedislike
ascend-robotascend-robot成员
3月19日 合入了pull request