From 876312f0b59b24f95704a37c93675e36a018a140 Mon Sep 17 00:00:00 2001 From: Nick Hill Date: Fri, 27 Feb 2026 07:54:24 -0800 Subject: [PATCH] [Core] Fix `gpu_worker.py` pre-commit errors (#35312) Signed-off-by: Nick Hill --- vllm/v1/worker/gpu_worker.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/vllm/v1/worker/gpu_worker.py b/vllm/v1/worker/gpu_worker.py index e35d0ef68..3aeb20839 100644 --- a/vllm/v1/worker/gpu_worker.py +++ b/vllm/v1/worker/gpu_worker.py @@ -744,7 +744,8 @@ class Worker(WorkerBase): # Create the profiler wrapper only on the first start call if self.profiler is None: - if self.profiler_config.profiler == "torch": + profiler_type = self.profiler_config.profiler + if profiler_type == "torch": self.profiler = TorchProfilerWrapper( self.profiler_config, worker_name=trace_name, @@ -754,9 +755,12 @@ class Worker(WorkerBase): logger.debug( "Starting torch profiler with trace name: %s", trace_name ) - elif self.profiler_config.profiler == "cuda": + elif profiler_type == "cuda": self.profiler = CudaProfilerWrapper(self.profiler_config) logger.debug("Starting CUDA profiler") + else: + logger.warning("Unrecognized profiler: %s", profiler_type) + return self.profiler.start() else: # Profiler already initialized. Restart profiling but keep