[sgl] Fix mamba tracking calculation in spec dec (#22239)

This commit is contained in:
Tarushii Goel
2026-04-10 14:46:16 +08:00
committed by GitHub
parent 6d79c60995
commit 0334d4b7e8
@@ -588,7 +588,7 @@ class SchedulerOutputProcessorMixin:
actual_seq_len = req.seqlen - 1
if (
actual_seq_len // mamba_track_interval
!= (actual_seq_len - result.accept_length_per_req_cpu[i])
!= (actual_seq_len - result.accept_length_per_req_cpu[i] - 1)
// mamba_track_interval
):
req.mamba_next_track_idx = (