[misc] Fix setup.py condition to avoid AMD from being mistaken with CPU (vllm-project#13022)

Signed-off-by: kevin <[email protected]>
Signed-off-by: SzymonOzog <[email protected]>

Author: khluu

setup.py

@@ -48,8 +48,9 @@ def load_module_from_path(module_name, path):
         "so vLLM may not be able to run correctly", sys.platform)
     VLLM_TARGET_DEVICE = "empty"
 elif (sys.platform.startswith("linux") and torch.version.cuda is None
-      and os.getenv("VLLM_TARGET_DEVICE") is None):
-    # if cuda is not available and VLLM_TARGET_DEVICE is not set,
+      and os.getenv("VLLM_TARGET_DEVICE") is None
+      and torch.version.hip is None):
+    # if cuda or hip is not available and VLLM_TARGET_DEVICE is not set,
     # fallback to cpu
     VLLM_TARGET_DEVICE = "cpu"