From 973c9d01da863cac9c51e8a5c0d390fc84b84fbc Mon Sep 17 00:00:00 2001 From: Victor Ziliang Peng Date: Sat, 13 Sep 2025 11:28:38 -0700 Subject: [PATCH] [Minor] Simplify duplicative device check for cuda (#24793) Signed-off-by: Ziliang Peng --- vllm/platforms/cuda.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/vllm/platforms/cuda.py b/vllm/platforms/cuda.py index 42fff2f07a0a..8e3436a9e73c 100644 --- a/vllm/platforms/cuda.py +++ b/vllm/platforms/cuda.py @@ -64,8 +64,7 @@ class CudaPlatformBase(Platform): if self.has_device_capability(80): # Ampere and Hopper or later NVIDIA GPUs. return [torch.bfloat16, torch.float16, torch.float32] - elif (not self.has_device_capability(80) - ) and self.has_device_capability(60): + if self.has_device_capability(60): # Pascal, Volta and Turing NVIDIA GPUs, BF16 is not supported return [torch.float16, torch.float32] # Kepler and Maxwell NVIDIA GPUs, only FP32 is supported,