From 8c8ebeb941d7bcc380b4aa685fd5201100c9eeca Mon Sep 17 00:00:00 2001 From: "Chendi.Xue" Date: Thu, 29 Jan 2026 11:56:30 -0600 Subject: [PATCH] [BUGFIX][XPU] fix memory check after XPU reuse GPU_worker (#33358) Signed-off-by: Chendi Xue --- vllm/v1/worker/gpu_worker.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/vllm/v1/worker/gpu_worker.py b/vllm/v1/worker/gpu_worker.py index dc70ce5c7..d9411387e 100644 --- a/vllm/v1/worker/gpu_worker.py +++ b/vllm/v1/worker/gpu_worker.py @@ -330,7 +330,7 @@ class Worker(WorkerBase): free_gpu_memory = profile_result.after_profile.free_memory # NOTE(woosuk): Here we assume that the other processes using the same # GPU did not change their memory usage during the profiling. - assert self.init_snapshot.free_memory > free_gpu_memory, ( + assert self.init_snapshot.free_memory >= free_gpu_memory, ( "Error in memory profiling. " f"Initial free memory {format_gib(self.init_snapshot.free_memory)} GiB, " f"current free memory {format_gib(free_gpu_memory)} GiB. "