From 1e7bf7970a1b87d7fc2acb64331e47b9990cd1ac Mon Sep 17 00:00:00 2001 From: Woosuk Kwon Date: Wed, 12 Mar 2025 23:38:12 -0700 Subject: [PATCH] minor Signed-off-by: Woosuk Kwon --- vllm/v1/core/sched/scheduler.py | 3 ++- vllm/v1/core/sched/simple_scheduler.py | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/vllm/v1/core/sched/scheduler.py b/vllm/v1/core/sched/scheduler.py index f5caff10bb10d..458abedbe71ba 100644 --- a/vllm/v1/core/sched/scheduler.py +++ b/vllm/v1/core/sched/scheduler.py @@ -617,7 +617,8 @@ class Scheduler(SchedulerInterface): def add_request(self, request: Request) -> None: self.waiting.append(request) self.requests[request.request_id] = request - record_queued(request) + if self.log_stats: + record_queued(request) def finish_requests( self, diff --git a/vllm/v1/core/sched/simple_scheduler.py b/vllm/v1/core/sched/simple_scheduler.py index 010e9f4575a47..f4a6728cfcef6 100644 --- a/vllm/v1/core/sched/simple_scheduler.py +++ b/vllm/v1/core/sched/simple_scheduler.py @@ -351,7 +351,8 @@ class SimpleScheduler(SchedulerInterface): def add_request(self, request: Request) -> None: self.waiting.append(request) self.requests[request.request_id] = request - record_queued(request) + if self.log_stats: + record_queued(request) def finish_requests( self,