文件最后提交记录最后更新时间
[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!315332 个月前