[Fix] Land the decode mamba checkpoint depth on the tree page under DCP (#35412)

Co-authored-by: Claude Opus 5 (1M context) <noreply@anthropic.com>
Co-authored-by: Ke Bao <ispobaoke@gmail.com>
This commit is contained in:
Khoa Pham
2026-08-20 12:15:42 -07:00
committed by GitHub
co-authored by Claude Opus 5 Ke Bao
parent 308bc1228b
commit eac91ac362
9 changed files with 60 additions and 19 deletions
@@ -198,8 +198,8 @@ class TestNgramMambaVerifyUpdate(CustomTestCase):
"sglang.srt.speculative.spec_utils.mambaish_config",
return_value={"some": "config"},
), patch(
"sglang.srt.speculative.spec_utils.get_exec",
return_value=MagicMock(mamba=MagicMock(mamba_track_interval=256)),
"sglang.srt.speculative.spec_utils.mamba_track_grid",
return_value=256,
):
commit_mamba_states_after_verify(
target_worker,