【合入来源】

(如有)请关联需求文档/issue链接
关联 issue: #1911 （https://gitcode.com/Ascend/pytorch/issues/1911）

• 需求
• 问题单
• issue/工单
• 重构优化
• 资料更新

【修改方案】

1. torch_npu/utils/_graph_tree.py — 新增 get_manager monkey-patch
   上游 torch._inductor.output_code.maybe_handle_backward_generation 和 torch._dynamo.backends.cudagraphs 会直接调用 torch._inductor.cudagraph_trees.get_manager，依赖它返回真实的 manager 实例来推进 backward 的 cudagraph 状态机。NPU 把自己的 NPUGraphTreeManager 注册在独立 registry（torch_npu.npu._graph_tree），导致上游那两条路径在 NPU 设备上拿不到 manager，触发 AssertionError 或 AttributeError。在 _apply_npugraph_tree_methods() 末尾把 torch._inductor.cudagraph_trees.get_manager 重指到 NPU 的 get_manager（签名一致、duck-typing 兼容），打通 backward 状态机闭环。

2. torch_npu/_inductor/codegen/wrapper.py — wrapper 体系重构以支持 graph partition
   提取 _NPUKernelCodegenMixin 混入类，把 NPU 特化逻辑（define_kernel 中 user_autotune → user_autotune_npu、PrecomputedGrid → PrecomputedGridNpu、FixedGrid → FixedGridNpu 替换；get_next_kernel_suffix、make_buffer_free 等）从原 NPUWrapperCodeGen 上移；新增 NPUSubgraphWrapperCodegen(_NPUKernelCodegenMixin, SubgraphPythonWrapperCodegen) 让 subgraph wrapper 共享 NPU 适配；NPUWrapperCodeGen.create() 在 is_subgraph=True 时返回新的 subgraph wrapper。

3. torch_npu/_inductor/lowering_op_list.py — graph partition 所需算子注册
   GENERATE_LIST 新增 prims.device_put、aten.unbind、torch.ops.higher_order.cond。

4. torch_npu/csrc/core/npu/NPUHooksInterface.{h,cpp} — Pinned memory 接口实现
   override 上游 at::PrivateUse1HooksInterface 新增的虚函数：isPinnedPtr() 走 CachingHostAllocator_isPinned，getPinnedMemoryAllocator() 返回 getPinnedMemoryAllocator()。同时新增 CachingHostAllocator.h 包含。

5. test/_inductor/test_inductor_graph_partition.py — graph partition 测试集
   新增 graph partition 场景的系列用例：dynamic shapes、condition op、custom op 拆分、subgraph wrapper user_autotune 兜底、forward cudagraph + backward fallback 等。

【资料变更】

不涉及

【接口变更】

不涉及。C++ 层 NPUHooksInterface::isPinnedPtr 和 getPinnedMemoryAllocator 是对上游 PrivateUse1HooksInterface 已有虚函数的 override 实现，非新增对外 API。

【功能验证】

在 test/_inductor/test_inductor_graph_partition.py 中新增覆盖 graph partition 场景的用例：

• test_graph_partition_dynamic_shapes：动态 shape 下产生 3 个 npugraph
• test_graph_partition_condition_op：cond / higher-order op 切图
• test_graph_partition_custom_op：cudagraph_unsafe 自定义算子切分边界
• test_graph_partition_subgraph_wrapper_user_autotune：子图 wrapper 的 user_autotune 替换路径
• forward cudagraph + backward fallback 的混合路径

通过 python test/_inductor/test_inductor_graph_partition.py 在 NPU 设备上跑通全部用例。

【CheckList】

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

• 代码注释完备，正确记录错误日志
• 代码实现进行了返回值、空指针等校验
• PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等
• PR持续集成流水线（CI）执行通过，代码检查无异常