From 7b09611d6bc5c09f0b311263dba9d30aca3d0f88 Mon Sep 17 00:00:00 2001
From: chenjian <1435317881@qq.com>
Date: Sat, 23 Aug 2025 11:55:53 +0800
Subject: [PATCH] [Bug fix] Fix batched token condition (#3565)

---
 fastdeploy/scheduler/dp_scheduler.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/fastdeploy/scheduler/dp_scheduler.py b/fastdeploy/scheduler/dp_scheduler.py
index d5d1d3967..65954b3c0 100644
--- a/fastdeploy/scheduler/dp_scheduler.py
+++ b/fastdeploy/scheduler/dp_scheduler.py
@@ -156,12 +156,12 @@ class DPLocalScheduler(LocalScheduler):
                         required_total_blocks += required_input_blocks + reserved_output_blocks
                         if required_total_blocks > available_blocks:
                             break
-                        if current_prefill_tokens > max_num_batched_tokens:
-                            break
 
                         requests.append(request.raw)
                         self.ids_read_cursor += 1
                         start_batch_time = time.time()
+                        if current_prefill_tokens > max_num_batched_tokens:
+                            break
                         if len(requests) >= batch:
                             break
                 if (