From 3b5aae278e7de6902bcca54b6b8ecb6b81fae86b Mon Sep 17 00:00:00 2001 From: YC Yen-Ching Tseng Date: Thu, 18 Jun 2026 05:54:33 +0800 Subject: [PATCH] Fix EagleDraftExtendInput missing kv_indptr crash with triton/DP attention (#28221) --- python/sglang/srt/speculative/eagle_info.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/python/sglang/srt/speculative/eagle_info.py b/python/sglang/srt/speculative/eagle_info.py index 73de4be2f..22b783dff 100644 --- a/python/sglang/srt/speculative/eagle_info.py +++ b/python/sglang/srt/speculative/eagle_info.py @@ -322,6 +322,10 @@ class EagleDraftExtendInput(SpecInput): num_tokens_per_req: int = -1 num_tokens_for_logprob_per_req: int = 1 + # None for draft-extend's idle batch; attention backends fall back to + # rebuilding plain metadata from seq_lens when this is None. + kv_indptr: torch.Tensor = None + def __post_init__(self): super().__init__(SpecInputType.EAGLE_DRAFT_EXTEND)