[Spec] Propagate state capture outputs in DFlash (#34026)

2026-08-07 14:42:30 -07:00
parent 9e3f6b746b
commit 8600457731
1 changed files with 2 additions and 0 deletions
@@ -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,
        )