| [feat][Inductor] Add MLIR support for v2.11.0
Co-authored-by: wangzili<wangzili14@huawei.com>
# message auto-generated for no-merge-commit merge:
!31533 merge mlir_master into master
[feat][Inductor] Add MLIR support for v2.11.0
Created-by: wangzili121
Commit-by: wangzili
Merged-by: ascend-robot
Description: <!--
PR描述模板更新日期:20260203
-->
# 【合入来源】
> <font color="red">**如有社区issue,请关联issue链接**</font>\
> <font color="red">**请勿携带内部流程信息(需求链接、问题单、内部issue等)**</font>
- [ ] 需求
- [ ] 问题单
- [ ] issue/工单
- [ ] 重构优化
- [ ] 资料更新
# 【修改方案】
- 新增ascend_npu_ir模块,实现MLIR后端的编译入口、算子映射和代码生成,提供三种启用方式:配置项、环境变量、torch.compile参数
- 在C++层实现MLIR绑定,新增torch_npu._C.mlir子模块提供核心接口
- 新增README文档,说明MLIR引入背景、使用方式和示例
# 【资料变更】
新增torch_npu/_inductor/ascend_npu_ir/README.md简要说明mlir的基本信息与使用方法
# 【接口变更】
新增以下客户面可见接口,详细使用方法已在 torch_npu/_inductor/ascend_npu_ir/README.md 中说明。
- 配置项:torch._inductor.config.npu_backend = "mlir"
- 环境变量:TORCHINDUCTOR_NPU_BACKEND='mlir'
- torch.compile参数:options={"npu_backend":"mlir"}
# 【功能验证】
已运行din/dcn/hllm/hf_Albert/BERT_pytorch五张网络验证能跑通精度没问题
# 【CheckList】
> PR提交人对以下CheckList自检项进行全量自检,自检通过或不涉及,均修改 [ ] 为 [x]
- [x] 代码注释完备,正确记录错误日志
- [x] 代码实现进行了返回值、空指针等校验
- [x] PR标题正确使用类型标签,如:feat、fix、refactor、docs、test等
- [x] PR持续集成流水线(CI)执行通过,代码检查无异常
See merge request: Ascend/pytorch!31533 | 2 个月前 |