执行摘要

• 一句话：集中 Mooncake PG 配置到传输引擎模块
• 推荐动作：值得精读的设计决策，特别是将依赖集中并通过初始化顺序保证共享实例的模式。建议关注 _resolve_backend 的简化以及 TE 初始化时机的选择。

功能与动机

Mooncake PG bootstrap configuration used to span across various modules: init_torch_distributed, init_distributed_environment, etc. This PR puts them inside mooncake_transfer_engine.py cleanly.

实现拆解

1. 调整 ModelRunner 初始化顺序（model_runner.py）：将设备设置（set_device）提前，并在 init_torch_distributed 之前调用 init_shared_mooncake_transfer_engine，确保共享 TransferEngine 可被 Mooncake PG 后端使用。
2. 简化 _resolve_backend（bootstrap.py）：移除 IB 设备过滤与 set_device_filter 调用，仅返回 'mooncake' 字符串；同时从 init_torch_distributed 中移除 set_device（移至步骤 1）。
3. 扩展 Mooncake TE 初始化条件（mooncake_transfer_engine.py）：在 maybe_init_shared_mooncake_transfer_engine 中添加 or server_args.elastic_ep_backend == 'mooncake' 条件；初始化成功后若需 Elastic EP，则调用 set_transfer_engine 注册全局共享引擎，并附带版本检查。
4. 清理重复的 EP 配置（parallel_state.py）：从 init_distributed_environment 中移除 mooncake 的导入和 set_host_ip 调用，避免重复初始化。

关键文件：

• python/sglang/srt/distributed/bootstrap.py（模块 引导模块；类别 source；类型 dependency-wiring；符号 _resolve_backend）: 核心入口 init_torch_distributed 与 _resolve_backend 被大幅度简化，设备设置逻辑移出。
• python/sglang/srt/model_executor/model_runner.py（模块 模型运行器；类别 source；类型 data-contract）: 修改了 init 中的初始化顺序，将 set_device 和 TE 初始化提前至 torch distributed 之前。
• python/sglang/srt/distributed/device_communicators/mooncake_transfer_engine.py（模块 传输引擎；类别 source；类型 dependency-wiring）: 集中了 Mooncake PG 配置，新增 Elastic EP 分支下的 set_transfer_engine 调用。
• python/sglang/srt/distributed/parallel_state.py（模块 并行状态；类别 source；类型 dependency-wiring）: 从 init_distributed_environment 中删除重复的 mooncake 配置代码。

关键符号：init_torch_distributed, _resolve_backend, maybe_init_shared_mooncake_transfer_engine, init_distributed_environment, ModelRunner.init

评论区精华

1. set_device 兼容性：gemini-code-assist[bot] 指出直接调用 set_device 可能在 CPU/MPS 上引发 AttributeError，建议先检查 hasattr。作者回复此段代码从 bootstrap.py 原封迁移，认为 CI 通过即可安全，最终未采纳该建议。
2. set_transfer_engine 导入健壮性：gemini-code-assist[bot] 建议对 set_transfer_engine 的导入使用 try-except 包裹以兼容旧版 mooncake。该建议被采纳，在最终代码中已实现。

• set_device 兼容性检查 (correctness): 未采纳，保持直接调用方式。
• set_transfer_engine 导入异常处理 (correctness): 已采纳，在最终代码中实现。

风险与影响

• 风险：
  1. 核心初始化路径变更：model_runner 中 set_device 与 TE 初始化移到 torch distributed 之前，若 TE 初始化失败将阻塞整个启动流程。
  2. 非 CUDA 设备兼容性：移出的 set_device 未添加 hasattr 检查，若在 CPU/NPU 等平台缺少 set_device 属性可能引发 AttributeError，但作者认为与原有行为一致。
  3. 缺少测试覆盖：本次改动未包含对应测试文件，回归风险由现有 CI 覆盖。
     - 影响：正向影响：Mooncake 配置集中化，降低后续维护成本；初始化逻辑更清晰。负向影响：需 mooncake-transfer-engine >= 0.3.11 以支持 set_transfer_engine；对非 Mooncake 用户无影响。
     - 风险标记：核心路径变更, 缺少测试覆盖, 兼容性风险

关联脉络

• 暂无明显关联 PR