mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2025-12-24 13:28:13 +08:00
Fix schedule error in splitwise deployment (#5149)
This commit is contained in:
@@ -514,6 +514,8 @@ class ResourceManagerV1(ResourceManager):
|
||||
error_reqs: list[tuple[str, str]] = []
|
||||
token_budget = self.config.scheduler_config.max_num_batched_tokens
|
||||
|
||||
self.check_and_free_block_tables()
|
||||
|
||||
# First, schedule the RUNNING requests.
|
||||
req_index = 0
|
||||
num_decoding_req_nums = 0
|
||||
|
||||
@@ -285,7 +285,7 @@ class LocalScheduler:
|
||||
if short_partial_requests + long_partial_requests > self.max_num_partial_prefills:
|
||||
break
|
||||
else:
|
||||
if current_prefill_tokens > max_num_batched_tokens:
|
||||
if current_prefill_tokens > max_num_batched_tokens and len(requests) > 0:
|
||||
break
|
||||
requests.append(request.raw)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user