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

safe changes from new dtype branch [pr] (#7397)

Browse Source 
* safe changes from new dtype branch [pr]

* only image test on GPU
This commit is contained in:
George Hotz
2024-10-30 16:18:48 +07:00
committed by GitHub
parent 0ca241693b
commit 4e2895f8d2
9 changed files with 36 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
test/test_linearizer_failures.py
View File

@@ -13,8 +13,8 @@ from tinygrad.shape.shapetracker import ShapeTracker
from tinygrad.shape.view import View
def helper_test_lin(lin: Kernel, opts, failed_platforms, rtol=1e-2, atol=1e-2):
if any(b.dtype == dtypes.half for b in lin.membufs) and not is_dtype_supported(dtypes.half): return
if any(b.dtype == dtypes.bfloat16 for b in lin.membufs) and not is_dtype_supported(dtypes.bfloat16): return
if any(b.dtype.base == dtypes.half for b in lin.membufs) and not is_dtype_supported(dtypes.half): return
if any(b.dtype.base == dtypes.bfloat16 for b in lin.membufs) and not is_dtype_supported(dtypes.bfloat16): return
for opt in opts:
try: