Skip to content

Commit

Permalink
Remove nvml-specific exception check in platform determination
Browse files Browse the repository at this point in the history
  • Loading branch information
conroy-cheers committed Oct 29, 2024
1 parent da79e3e commit ef7ff3c
Showing 1 changed file with 1 addition and 5 deletions.
6 changes: 1 addition & 5 deletions vllm/platforms/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand All @@ -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

Expand Down

0 comments on commit ef7ff3c

Please sign in to comment.