From ef7ff3c80920ad91923e1b4e906d9901312610e5 Mon Sep 17 00:00:00 2001
From: ccheers <conroy@corncheese.org>
Date: Tue, 29 Oct 2024 13:58:06 +1100
Subject: [PATCH] Remove nvml-specific exception check in platform
 determination

---
 vllm/platforms/__init__.py | 6 +-----
 1 file changed, 1 insertion(+), 5 deletions(-)

diff --git a/vllm/platforms/__init__.py b/vllm/platforms/__init__.py
index 41fe79d1dd63f..805694450e47a 100644
--- a/vllm/platforms/__init__.py
+++ b/vllm/platforms/__init__.py
@@ -27,9 +27,7 @@
             is_cuda = True
     finally:
         pynvml.nvmlShutdown()
-except ImportError:
-    pass
-except pynvml.NVMLError_LibraryNotFound:
+except Exception:
     # CUDA is supported on Jetson, but NVML is not.
     import os
 
@@ -40,8 +38,6 @@ def cuda_is_jetson() -> bool:
     if cuda_is_jetson():
         is_cuda = True
         is_cuda_jetson = True
-except Exception:
-    pass
 
 is_rocm = False