From 86c5617787fb14c9797319ea3d7a4a96cc913295 Mon Sep 17 00:00:00 2001
From: Chenxi Li <41864925+ConnorLi96@users.noreply.github.com>
Date: Wed, 4 Mar 2026 21:13:21 -0800
Subject: [PATCH] [BUG]: fix prevent illegal memory access in Mamba SSM
 tracking during EAGLE speculative verification (#19415)

Co-authored-by: ConnorLi96 <ConnorLi96@users.noreply.github.com>
---
 python/sglang/srt/speculative/eagle_info.py    | 2 ++
 python/sglang/srt/speculative/eagle_info_v2.py | 2 ++
 2 files changed, 4 insertions(+)

diff --git a/python/sglang/srt/speculative/eagle_info.py b/python/sglang/srt/speculative/eagle_info.py
index ac629c7ee..c2f90c378 100644
--- a/python/sglang/srt/speculative/eagle_info.py
+++ b/python/sglang/srt/speculative/eagle_info.py
@@ -154,6 +154,8 @@ class EagleVerifyInput(SpecInput, EagleVerifyInputV2Mixin):
                 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
index 0ae6e3906..d96c89da5 100644
--- a/python/sglang/srt/speculative/eagle_info_v2.py
+++ b/python/sglang/srt/speculative/eagle_info_v2.py
@@ -242,6 +242,8 @@ class EagleVerifyInputV2Mixin:
                     dtype=torch.int64,
                     device=device,
                 )
+                batch.mamba_track_mask = None
+                batch.mamba_track_seqlens = None
 
         # Get a forward batch
         batch.forward_mode = (