From 54f7c3e45863671e77cf1b327c3bc71fd9a2cd2f Mon Sep 17 00:00:00 2001 From: Michael Goin Date: Wed, 22 Jul 2026 20:03:15 -0600 Subject: [PATCH] Update cudagraph_utils.py Signed-off-by: Michael Goin --- vllm/v1/worker/gpu/cudagraph_utils.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/vllm/v1/worker/gpu/cudagraph_utils.py b/vllm/v1/worker/gpu/cudagraph_utils.py index 068008b9acb..591e91be49f 100644 --- a/vllm/v1/worker/gpu/cudagraph_utils.py +++ b/vllm/v1/worker/gpu/cudagraph_utils.py @@ -522,8 +522,7 @@ class ModelCudaGraphManager(CudaGraphManager): batch_descriptor=batch_descriptor, is_padding=input_buffers.is_padding[:num_tokens], skip_attention=( - cg_mode == CUDAGraphMode.PIECEWISE - and not self.use_breakable_cg + cg_mode == CUDAGraphMode.PIECEWISE and not self.use_breakable_cg ), ): if cg_mode == CUDAGraphMode.PIECEWISE: