From 696b6531931803f0e31b5c56a386dda1bc98db84 Mon Sep 17 00:00:00 2001 From: Woosuk Kwon Date: Wed, 10 Apr 2024 08:02:21 +0000 Subject: [PATCH] yapf --- setup.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/setup.py b/setup.py index 616324e812bcd..476e225a3a1aa 100644 --- a/setup.py +++ b/setup.py @@ -189,8 +189,7 @@ class cmake_build_ext(build_ext): def _is_cuda() -> bool: has_cuda = torch.version.cuda is not None - return (VLLM_TARGET_DEVICE == "cuda" - and has_cuda + return (VLLM_TARGET_DEVICE == "cuda" and has_cuda and not (_is_neuron() or _is_tpu()))