已合并
【rfc】mlir_triton_lowering_file_refactoring #37279
cuiduo创建于 6月1日
【rfc】mlir_triton_lowering_file_refactoring #37279
已合并
cuiduo创建于 6月1日
cuiduo
cuiduo成员
6月1日

【合入来源】

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

【修改方案】

对lowering部分进行重构,torch_npu/_inductor 在 Lowering 层 的多后端(Triton / MLIR / DVM)共享重构方案及转测要求。核心改动是:将 Triton 与 MLIR/DVM 两套 lowering 中重复的逻辑抽取到 lowering_common.py,通过 lowering_patch.py 实现 同进程 backend 切换时的 lowering 快照/恢复,并配合 _load_backend() 按需加载各后端 lowering 实现,消除原先 inductor_patch/init.py import 时整表 monkey-patch 的副作用。详见关联文档

【资料变更】

不涉及

【接口变更】

不涉及

【功能验证】

已通过

【CheckList】

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

likedislike
Pull Request已成功合入, 合并人@ascend-robot
(感谢 cuiduo 的贡献)
cuiduocuiduo成员
6月1日 创建了 pull request,commit 3f851139
openLiBingCI成员6月1日进行代码检视1
torch_npu/_inductor/lowering_patch.py
@@ -0,0 +136,6 @@
136+ )
137+ 
138+ # Ensure IR patches (TracedGraph hooks) are registered.
139+ import torch_npu._inductor.ascend_npu_ir.ascend_npu_ir.npu.inductor_patch.ir # noqa: F401
140+ 
141+ apply_mlir_lowering_patch(npu_lowering)
142+ _patch_scheduler()
openLiBingCI6月1日评论:

此条代码评论区间+136+175

【openlibing.ci】识别到代码检查告警抑制注释,匹配工具:flake8,请Committer检视其合理性。

likedislike
openLiBingCI成员6月1日进行代码检视1
torch_npu/_inductor/lowering_patch.py
@@ -0,0 +136,6 @@
136+ )
137+ 
138+ # Ensure IR patches (TracedGraph hooks) are registered.
139+ import torch_npu._inductor.ascend_npu_ir.ascend_npu_ir.npu.inductor_patch.ir # noqa: F401
140+ 
141+ apply_mlir_lowering_patch(npu_lowering)
142+ _patch_scheduler()
openLiBingCI6月1日评论:

此条代码评论区间+136+175

【openlibing.ci】识别到代码检查告警抑制注释,匹配工具:flake8,请Committer检视其合理性。

likedislike
ascend-robotascend-robot成员
6月1日 添加了label:ascend-cla/yes
此处折叠了324条消息 查看更多
ascend-robotascend-robot成员
6月9日 添加了label:approved
ZhihaoLi
ZhihaoLi成员
6月10日 评论:

/approve

likedislike
ascend-robotascend-robot成员
6月10日 添加了label:lgtm
ascend-robotascend-robot成员
6月10日 合入了pull request
ascend-robot
ascend-robot成员
6月10日 评论:
流水线 pytorch_gitcode_PR_multiVersion#10407 [ commitID:bceff4a8 ] 已完成
likedislike