Support multi-step alignment and pipeline integration in dump comparator (#19378)

2026-02-26 10:23:22 +08:00
parent 4e843f1216
commit 265eb56d44
22 changed files with 535 additions and 41 deletions
@@ -2127,7 +2127,9 @@ class TestRegisterForwardHook:
 class TestPluginCoreFields:
    def test_sglang_core_fields(self):
        plugin = _SGLangPlugin()
-        assert plugin.core_fields() == frozenset({"input_ids", "positions", "seq_lens"})
+        assert plugin.core_fields() == frozenset(
+            {"input_ids", "positions", "seq_lens", "req_pool_indices", "rids"}
+        )

    def test_megatron_core_fields(self):
        plugin = _MegatronPlugin()