github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-26 07:18:40 -05:00
Code Issues Packages Projects Releases Wiki Activity

use clang as default instead of llvm for cpu (#4035)

Browse Source 
llvm has problems with fp16 and comparing with nan
This commit is contained in:
chenyu
2024-04-02 00:02:18 -04:00
committed by GitHub
parent 5311b45053
commit 05e7f930ee
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

@@ -29,7 +29,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", "HSA", "CUDA", "GPU", "LLVM", "CLANG"]:
for device in ["METAL", "HSA", "CUDA", "GPU", "CLANG", "LLVM"]:
try:
if self[device]: return device
except Exception: pass