执行摘要

• 一句话：命名 VAE 解码并行轴常量，消除重复字符串
• 推荐动作：本 PR 虽小但体现了良好的代码习惯（提取常量、封装函数、保持测试与源码同步）。值得推荐为团队内部代码重构的示例。建议精读以了解 sglang 多模态推理中 VAE 解码的并行组划分设计。

功能与动机

PR 标题和提交信息指出要“document VAE decode parallel group axes”。原代码在 parallel_state.py 的第 461 行直接写死了 rank_generator.get_ranks("tp-sp-pp-cfg")，且在测试中也重复了相同的字符串。这样既不利于文档化（说明各个轴的含义和约束），也容易因后续修改不同步导致隐藏的 bug。通过提取常量和辅助函数，可以集中表达“VAE 解码可以在 TP/SP/PP/CFG 内分片但不能跨 DP”这一设计约束。

实现拆解

1. 定义常量 _VAE_DECODE_PARALLEL_AXES：在 parallel_state.py 第 72 行新增模块级字符串常量 "tp-sp-pp-cfg"，并添加注释说明该常量表示 VAE 解码可跨 TP、SP、PP、CFG 轴，但禁止跨 DP 轴。
2. 封装辅助函数 _get_vae_decode_group_ranks：新增函数，接受 RankGenerator 实例并返回 rank_generator.get_ranks(_VAE_DECODE_PARALLEL_AXES) 的结果，同时携带 docstring 解释设计约束。
3. 替换原有硬编码调用：将 init_parallel_group_coordinator 调用中原来的直接字符串 "tp-sp-pp-cfg" 替换为调用 _get_vae_decode_group_ranks(rank_generator)。
4. 更新测试用例：在 test_vae_spatial_parallel_decode.py 的 test_decode_rank_groups_cover_non_dp_parallel_axes 中，将之前的 rank_generator.get_ranks("tp-sp-pp-cfg") 改为调用 parallel_state._get_vae_decode_group_ranks(rank_generator)，使测试与源码共享同一来源。

关键文件：

• python/sglang/multimodal_gen/runtime/distributed/parallel_state.py（模块 分布式状态；类别 source；类型 core-logic；符号 _VAE_DECODE_PARALLEL_AXES, _get_vae_decode_group_ranks）: 核心源码文件，定义 VAE 解码并行轴常量和辅助函数，并替换了原有硬编码调用。
• python/sglang/multimodal_gen/test/unit/test_vae_spatial_parallel_decode.py（模块 VAE 测试；类别 test；类型 test-coverage）: 测试文件，将直接调用 RankGenerator.get_ranks 替换为调用 parallel_state._get_vae_decode_group_ranks，确保测试与源码同步。

关键符号：_get_vae_decode_group_ranks

关键源码片段

python/sglang/multimodal_gen/runtime/distributed/parallel_state.py

核心源码文件，定义 VAE 解码并行轴常量和辅助函数，并替换了原有硬编码调用。

def _get_vae_decode_group_ranks(
    rank_generator: RankGenerator,
) -> list[list[int]]:
    # VAE decode happens after each DP replica owns a different request result.
    # Decode can shard one request across TP/SP/PP/CFG ranks, but must not cross DP.
    return rank_generator.get_ranks(_VAE_DECODE_PARALLEL_AXES)

评论区精华

PR 未产生 review 评论，仅有作者 mickqian 在 Issue 中触发了一次 CI 重跑（/tag-and-rerun-ci）。此外 gemini-code-assist bot 回复了两次配额用尽的消息。所有人员（包括合并者）均为 mickqian，该 PR 是单人变更且未引发技术讨论。

• 暂无高价值评论线程

风险与影响

• 风险：
  1. 回归风险低：变更将硬编码字符串替换为等价常量调用，逻辑完全等价；测试也同步更新，CI 已通过。
  2. 兼容性：不改变任何公共 API 或配置键，对用户无感知。
  3. 性能：无影响，仅引入一个模块级字符串和一个简单函数调用。
     - 影响：影响范围仅限于 VAE 空间并行解码功能的内部初始化代码和对应单元测试。对用户无直接功能影响，但提升了内部一致性和可维护性。后续若有开发者需要修改 VAE 解码的并行轴，只需修改 _VAE_DECODE_PARALLEL_AXES 常量一处。
     - 风险标记：暂无

关联脉络

• PR #29519 [diffusion] warmup: default to model sampling resolution (declare Z-Image default): 同属 diffusion 模块的近期 PR，与本 PR 均涉及 diffusion 相关配置和测试。