Fix extra_buffer_lazy guard bypass (#31942)

2026-07-21 20:41:35 +08:00
parent e4eea7ce2f
commit 0a06cc5317
1 changed files with 5 additions and 7 deletions
@@ -4822,14 +4822,12 @@ class ServerArgs:
                "extra_buffer_lazy unsupported under PD disaggregation; use "
                "--mamba-radix-cache-strategy extra_buffer."
            )
+            algo = (view.speculative_algorithm or "").upper()
+            assert algo not in ("DFLASH", "DSPARK"), (
+                f"extra_buffer_lazy unsupported with {view.speculative_algorithm}; "
+                "use --mamba-radix-cache-strategy extra_buffer."
+            )
        if view.speculative_num_draft_tokens is not None:
-            if view.mamba_radix_cache_strategy == "extra_buffer_lazy":
-                # The dflash family's verify bypasses prepare_mamba_track_for_verify.
-                assert view.speculative_algorithm not in ("DFLASH", "DSPARK"), (
-                    f"extra_buffer_lazy unsupported with "
-                    f"{view.speculative_algorithm}; use "
-                    "--mamba-radix-cache-strategy extra_buffer."
-                )
            assert view.mamba_track_interval >= view.speculative_num_draft_tokens
        if view.page_size is not None:
            assert view.mamba_track_interval % view.page_size == 0