执行摘要

该PR修复了比较器端到端测试中的两个关键缺陷：添加缺失的polars依赖解决了测试提前崩溃问题，并修正了dp-attention测试中的维度标注和步骤限制逻辑。这些修复确保了测试能够实际运行并正确验证功能，提升了测试基础设施的可靠性，对系统运行时无直接影响。

功能与动机

修复比较器端到端测试中的长期缺陷：由于比较器模块(sglang.srt.debug_utils.comparator)在顶层导入polars，但该依赖从未添加到任何依赖组中，导致测试在运行比较前就因ModuleNotFoundError崩溃。同时，dp-attention测试中存在逻辑错误：mlp_output的维度被错误标注为已全部归约，而实际上在转储点仍为部分状态；且解码步骤的令牌分布在多个DP rank上，违反了单rank假设。这些问题在之前的dump比较器PR系列（#19274, #19681）中引入但一直未被发现，因为依赖缺失导致测试从未真正执行比较。

实现拆解

实现分为两个关键文件修改：

1. 依赖修复（python/pyproject.toml）：
   

   test = [
     "pandas",
     "parameterized",
     "peft>=0.18.0",
   + "polars",
     "pytest",
     "pytest-cov",
     "diff-cover",
   ]
   

   
   添加polars到test依赖组，确保比较器模块能正常导入。

2. 测试逻辑修正（test/registered/debug_utils/test_engine_dumper_comparator_e2e.py）：

   • 将mlp_output的dims从't h # tp:replicated'改为't h[moe_tp:partial] # tp:replicated'，反映MLP输出在转储点仍为部分状态（归约分散在postprocess_layer()中发生）。
   • 在test_dp_attention中添加extra_comparator_args参数：
     

     extra_comparator_args=[
         "--end-step",
         "0",
         "--allow-failed-pattern",
         "mlp_output",
     ]
     

     
     限制比较到步骤0（预填充），避免解码步骤中多DP rank导致的比较错误；并允许mlp_output比较失败，因为FusedMoE调度器组合路径可能包含隐式全部归约。

评论区精华

无review评论，但PR body中提供了详细的修复说明和测试验证链接：

这些bug在dump比较器PR系列（#19274, #19681）中引入但从未被发现，因为缺失的polars依赖在比较运行前就使比较器子进程崩溃。

作者通过多个CI运行链接（如https://github.com/sgl-project/sglang/actions/runs/23853524625/job/69540028112）验证了修复后测试通过。

风险与影响

风险：

• 依赖添加仅影响测试环境，无运行时风险。
• 测试逻辑修正使比较更准确，但允许mlp_output失败可能掩盖FusedMoE调度器的潜在问题。
• 限制比较到预填充步骤可能遗漏解码步骤的回归，需后续测试补充。

影响：

• 直接影响测试基础设施，确保比较器端到端测试能正常运行，有助于发现未来代码变更引入的回归。
• 对用户和系统无直接影响，但提升了测试覆盖的可靠性。

关联脉络

该PR与历史PR #19274和#19681关联，这些PR引入了dump比较器功能但未正确处理依赖和测试逻辑。从近期PR分析看，该仓库频繁进行测试基础设施的维护和修复（如#21873添加网络超时、#21830修复CI稳定性），表明团队重视测试可靠性。本次修复延续了这一趋势，解决了长期存在的测试缺陷，为后续功能开发提供了更稳定的验证基础。