Pull Request已成功合入, 合并人@ascend-robot
(感谢 lv-kaimeng 的贡献)变更摘要
本 PR 为 bug 修复(fix master bug),主要围绕 torch_npu/_inductor 中与 FX 图导出/精度校验(INDUCTOR_ASCEND_CHECK_ACCURACY)相关的链路进行修正与补齐。核心改动包括:_patch_mark_reuse 透传可变关键字参数,确保 mark_reuse 调用与上游签名一致;NPUTritonKernel 在 inductor_meta 中重新注入 traced_graph_hash / traced_graph_dir 占位符,供精度校验工具定位导出的 FX 图;lowering.py 补齐 create_fake_input、subtract_graph、snodes_to_fx、generate_fx_graph_code 等符号的导入。
主要改动
-
_patch_mark_reuse支持**kwargs透传:torch_npu/_inductor/codegen/ir_fx.py中_patch_mark_reuse(self, users, **kwargs)将额外的关键字参数传递给self.data.mark_reuse(users, **kwargs)(有/无traced_graph两个分支均如此),避免调用签名不匹配导致的兼容性问题。 -
NPUTritonKernel重新注入 FX 图定位元数据:torch_npu/_inductor/codegen/triton.py在create_inductor_meta中通过setdefault将traced_graph_hash、traced_graph_dir占位符写回inductor_meta,因为上游super().codegen_kernel()重写inductor_meta时会丢弃这些字段;实际值由SchedulerNodeCodeGenerator.define_kernel在启用dump_fx_graph时文本替换。 -
补齐
lowering.py缺失的符号导入:torch_npu/_inductor/lowering.py新增从.lowering_common导入create_fake_input、subtract_graph,从.lowering_fx导入create_fx_from_snodes_by_traced_graph、create_compile_kwargs、generate_fx_graph_code、dump_fx_graph_code、snodes_to_fx,修复相关函数引用导致的运行时错误。




【合入来源】
【修改方案】
【资料变更】
【接口变更】
【功能验证】
【CheckList】