From 41f3884438c082c4cc2250eb800b8586e1a103d8 Mon Sep 17 00:00:00 2001 From: quanliu <33453350+quanliu1991@users.noreply.github.com> Date: Mon, 13 Oct 2025 09:25:42 +0800 Subject: [PATCH] [Bugfix][Core]Fix block table out-of-range issue in priority scheduling (#26661) Signed-off-by: quanliu <18646313696@163.com> --- vllm/v1/core/sched/scheduler.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/vllm/v1/core/sched/scheduler.py b/vllm/v1/core/sched/scheduler.py index 32c2eb8a46526..16808417766ae 100644 --- a/vllm/v1/core/sched/scheduler.py +++ b/vllm/v1/core/sched/scheduler.py @@ -271,6 +271,9 @@ class Scheduler(SchedulerInterface): self.running.remove(preempted_req) if preempted_req in scheduled_running_reqs: scheduled_running_reqs.remove(preempted_req) + token_budget += num_scheduled_tokens[preempted_req.request_id] + req_to_new_blocks.pop(preempted_req.request_id) + num_scheduled_tokens.pop(preempted_req.request_id) else: preempted_req = self.running.pop()