commaai
/
tinygrad
mirror of https://github.com/commaai/tinygrad.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

hotfix: swap HIP/CUDA bringup order to prevent delay on tinybox

This commit is contained in:
George Hotz 2024-02-09 18:41:25 +01:00
parent 28ba1c5406
commit dc82ef6660
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tinygrad/device.py
View File

@ -30,7 +30,7 @@ class _Device:
def DEFAULT(self) -> str:
device_from_env: Optional[str] = functools.reduce(lambda val, ele: ele if getenv(ele) == 1 else val, self._devices, None) # type: ignore
if device_from_env: return device_from_env
for device in ["METAL", "CUDA", "HIP", "GPU"]:
for device in ["METAL", "HIP", "CUDA", "GPU"]:
try:
if self[device]: return device
except Exception: pass