From d379bda4fadef05a06a2609fdd482ad3d04031ea Mon Sep 17 00:00:00 2001
From: Shangming Cai <caishangming@linux.alibaba.com>
Date: Tue, 8 Jul 2025 17:49:32 +0800
Subject: [PATCH] [Bugfix] Fix two batch overlap with auto DeepEP Dispatch
 (#7853)

Signed-off-by: Shangming Cai <caishangming@linux.alibaba.com>
---
 python/sglang/srt/two_batch_overlap.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/python/sglang/srt/two_batch_overlap.py b/python/sglang/srt/two_batch_overlap.py
index 6d2d73028..fc419b03c 100644
--- a/python/sglang/srt/two_batch_overlap.py
+++ b/python/sglang/srt/two_batch_overlap.py
@@ -490,6 +490,7 @@ class TboForwardBatchPreparer:
         output_dict["spec_info"] = output_spec_info
         for key in [
             "forward_mode",
+            "is_extend_in_batch",
             "return_logprob",
             "req_to_token_pool",
             "token_to_kv_pool",