From dc82ef6660e9b8e45ecc2b7466336ba7ce200852 Mon Sep 17 00:00:00 2001 From: George Hotz Date: Fri, 9 Feb 2024 18:41:25 +0100 Subject: [PATCH] hotfix: swap HIP/CUDA bringup order to prevent delay on tinybox --- tinygrad/device.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tinygrad/device.py b/tinygrad/device.py index cb6143b972..46e528bb6f 100644 --- a/tinygrad/device.py +++ b/tinygrad/device.py @@ -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