xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2025-12-15 03:45:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Bugfix][Core]Fix block table out-of-range issue in priority scheduling (#26661)

Browse Source 
Signed-off-by: quanliu <18646313696@163.com>
This commit is contained in:
quanliu 2025-10-13 09:25:42 +08:00 committed by GitHub
parent 60e419c1ee
commit 41f3884438
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
vllm/v1/core/sched/scheduler.py
View File

@ -271,6 +271,9 @@ class Scheduler(SchedulerInterface):
self.running.remove(preempted_req) self.running.remove(preempted_req)
if preempted_req in scheduled_running_reqs: if preempted_req in scheduled_running_reqs:
scheduled_running_reqs.remove(preempted_req) 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: else:
preempted_req = self.running.pop() preempted_req = self.running.pop()