[BUG]: fix prevent illegal memory access in Mamba SSM tracking during EAGLE speculative verification (sgl-project#19415)

ConnorLi96 · web-flow · commit 86c5617787fb · 2026-03-04T21:13:21.000-08:00
Co-authored-by: ConnorLi96 &lt;ConnorLi96@users.noreply.github.com&gt;
diff --git a/python/sglang/srt/speculative/eagle_info.py b/python/sglang/srt/speculative/eagle_info.py
@@ -154,6 +154,8 @@ def prepare_for_verify(self, batch: ScheduleBatch, page_size: int):
                 dtype=torch.int64,
                 device=batch.device,
             )
+            batch.mamba_track_mask = None
+            batch.mamba_track_seqlens = None
 
     def generate_attn_arg_prefill(
         self,
diff --git a/python/sglang/srt/speculative/eagle_info_v2.py b/python/sglang/srt/speculative/eagle_info_v2.py
@@ -242,6 +242,8 @@ def prepare_for_v2_verify(
                     dtype=torch.int64,
                     device=device,
                 )
+                batch.mamba_track_mask = None
+                batch.mamba_track_seqlens = None
 
         # Get a forward batch
         batch.forward_mode = (

Original file line number	Diff line number	Diff line change
`@@ -154,6 +154,8 @@ def prepare_for_verify(self, batch: ScheduleBatch, page_size: int):`
`154`	`154`	`dtype=torch.int64,`
`155`	`155`	`device=batch.device,`
`156`	`156`	`)`
	`157`	`+ batch.mamba_track_mask = None`
	`158`	`+ batch.mamba_track_seqlens = None`
`157`	`159`
`158`	`160`	`def generate_attn_arg_prefill(`
`159`	`161`	`self,`
Original file line number	Diff line number	Diff line change
`@@ -242,6 +242,8 @@ def prepare_for_v2_verify(`
`242`	`242`	`dtype=torch.int64,`
`243`	`243`	`device=device,`
`244`	`244`	`)`
	`245`	`+ batch.mamba_track_mask = None`
	`246`	`+ batch.mamba_track_seqlens = None`
`245`	`247`
`246`	`248`	`# Get a forward batch`
`247`	`249`	`batch.forward_mode = (`