From 860045773171abd4183780c7d5d7a6c706bb87c9 Mon Sep 17 00:00:00 2001 From: Nan Jiang <59716405+nanjiangwill@users.noreply.github.com> Date: Fri, 7 Aug 2026 14:42:30 -0700 Subject: [PATCH] [Spec] Propagate state capture outputs in DFlash (#34026) --- python/sglang/srt/speculative/dflash_worker_v2.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/python/sglang/srt/speculative/dflash_worker_v2.py b/python/sglang/srt/speculative/dflash_worker_v2.py index 4f3c100b8..d8d94b1d4 100644 --- a/python/sglang/srt/speculative/dflash_worker_v2.py +++ b/python/sglang/srt/speculative/dflash_worker_v2.py @@ -1891,4 +1891,6 @@ class DFlashWorkerV2(BaseSpecWorker): # The non-overlap (sync) scheduler path advances batch.seq_lens # from the result; overlap carries it via next_draft_input instead. new_seq_lens=new_seq_lens, + routed_experts_output=target_out.routed_experts_output, + indexer_topk_output=target_out.indexer_topk_output, )