github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-23 05:48:08 -05:00
Code Issues Packages Projects Releases Wiki Activity

exclude half with PYTHON in test_dtype.is_dtype_supported (#3351)

Browse Source 
half memoryview only in 3.12+. rest of the test_dtype (bounty) seems to be legit issue in ops_python.
This commit is contained in:
chenyu
2024-02-08 20:10:25 -05:00
committed by GitHub
parent c32ea95d7d
commit 7c1c6efee5
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
test/test_dtype.py
View File

@@ -21,7 +21,8 @@ def is_dtype_supported(dtype: DType, device: str = Device.DEFAULT):
# for CI GPU, cl_khr_fp16 isn't supported
# for CI LLVM, it segfaults because it can't link to the casting function
# CUDA in CI uses CUDACPU that does not support half
if dtype == dtypes.half: return not (CI and device in ["GPU", "LLVM", "CUDA"])
# PYTHON supports half memoryview in 3.12+ https://github.com/python/cpython/issues/90751
if dtype == dtypes.half: return not (CI and device in ["GPU", "LLVM", "CUDA"]) and device != "PYTHON"
if dtype == dtypes.float64: return device != "METAL" and not (OSX and device == "GPU")
return True