github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-24 22:38:16 -05:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
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
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