变更摘要

此 PR 将 NPU 的 Dynamo/Inductor 编译器集成从 import torch_npu 时的立即初始化改为懒加载模式。核心思路是：import torch_npu 不再导入 torch._dynamo、torch._inductor、torch_npu._inductor 等重型模块，仅在用户实际调用 torch.compile() 或触发图捕获时才按需初始化。实现手段包括 setuptools entry points、meta_path finder、run_once 并发安全装饰器，以及多个子模块的 __getattr__ 懒加载。同时将 MarkStepBox 和 RNG prim 补丁等状态/逻辑拆分到独立轻量模块，避免级联导入。

主要改动

• Dynamo/Inductor 懒加载机制：新增 _DynamoPostImportFinder（meta_path finder）在 torch._dynamo 首次被导入时自动触发 _lazy_dynamo_setup()；新增 _lazy_inductor_setup() 仅在 Inductor 类后端被选用时才初始化 torch_npu._inductor；setup.py 新增 torch_dynamo_backends entry points（npu、npugraph_ex、npugraphs），由 Dynamo 框架在需要时回调。

• 注册时机调整与 run_once 并发安全：registry_manager._register_components() 中移除 _register_dynamo()，改为 _register_rng_prims()，Dynamo 后端注册延迟到 _lazy_dynamo_setup 中执行。run_once 装饰器增加 threading.Condition 并发等待、os.register_at_fork fork 后重置、失败时释放标记等机制，确保首次调用在多线程/fork 场景下正确。

• MarkStepBox 与 RNG prim 补丁拆分为独立模块：MarkStepBox 和 mark_step_begin() 从 torch_npu/npu/_graph_tree.py 移至新文件 torch_npu/npu/_graph_tree_state.py；_max_unpoolnd_patch、patch_philox_rand_offset 等 RNG 补丁从 torch_npu/utils/_inductor.py 移至新文件 torch_npu/utils/_rng_prims_patch.py，原 _inductor.py 仅保留重导出。两者均避免导入时拉入重型依赖。

• 子模块按需导入：torch_npu.npu.__init__.py 通过 __getattr__ 对 npugraph_ex 实现懒加载；torch_npu.distributed.__init__.py 通过 __getattr__ 对 fsdp 子模块实现懒加载；torch_npu/dynamo/__init__.py 中 _install_lazy_torchair() 按需注册 torchair 懒代理模块。

• transfer_to_npu.py 显式导入与 setattr 替换：将原先依赖 import torch_npu 副作用触发的 Dynamo/Inductor 模块导入改为显式 import 语句；将直接属性赋值（如 torch._inductor.utils.get_gpu_type = ...）改为 setattr 调用，避免触发模块的 __getattr__/描述符等副作用。

openLiBing可接管PR, PR详情链接:
https://www.openlibing.com/apps/transitionPrDetail?owner=Ascend&repo=pytorch&number=44294&platform=gitcode&codeHostingPlatformFlag=gitcode

🤖 AI Code Review

⚠️ 审查未能完成

本次代码审查未能完成，可能是临时性故障。

你可以在评论区输入 /ai review 重新发起审查。若多次失败，请联系管理员并附上下方追踪 ID。

追踪 ID: task-175411-run-168340