Skip to content

Commit

Permalink
Format
Browse files Browse the repository at this point in the history
  • Loading branch information
conroy-cheers committed Oct 27, 2024
1 parent b3117dc commit 3cc28e3
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions vllm/platforms/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -30,9 +30,11 @@
except pynvml.NVMLError_LibraryNotFound:
# CUDA is supported on Jetson, but NVML is not.
import os

def cuda_is_jetson() -> bool:
return os.path.isfile("/etc/nv_tegra_release") \
or os.path.exists("/sys/class/tegra-firmware")

if cuda_is_jetson():
is_cuda = True
is_cuda_jetson = True
Expand Down

0 comments on commit 3cc28e3

Please sign in to comment.