From a354ec9dadf1eae99be402902eae33ad21cfbb6a Mon Sep 17 00:00:00 2001 From: George Hotz Date: Wed, 3 Jan 2024 13:16:54 -0800 Subject: [PATCH] Revert "hotfix: HIP is the default device on HIP platforms" This reverts commit b748b569f52678d9b6b523dd29c0bf441fa900f7. --- tinygrad/device.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tinygrad/device.py b/tinygrad/device.py index 3b6c867895..8a6277afe4 100644 --- a/tinygrad/device.py +++ b/tinygrad/device.py @@ -28,7 +28,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._buffers, None) # type: ignore if device_from_env: return device_from_env - for device in ["METAL", "CUDA", "HIP", "GPU"]: + for device in ["METAL", "CUDA", "GPU"]: try: if self[device]: return device except Exception: pass