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

Revert "hotfix: HIP is the default device on HIP platforms"

Browse Source 
This reverts commit b748b569f5.
This commit is contained in:
George Hotz
2024-01-03 13:16:54 -08:00
parent b748b569f5
commit a354ec9dad
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

@@ -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