github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-29 03:00:14 -04:00
Code Issues Packages Projects Releases Wiki Activity

Fix linearizer failure 26 test (#3906)

Browse Source 
* Adjust adds between WHERE and PHI

* Not much better

* undo recursive change

* hm

* iterate over where, not factored op

* oo

* consts only for loop

* UNdo var name change

* update

---------

Co-authored-by: Patrick Tsai <patosai@users.noreply.github.com>
This commit is contained in:
Patrick Tsai
2024-03-24 16:34:13 -04:00
committed by GitHub
parent 10673d1447
commit e27129a798
2 changed files with 9 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/test_linearizer_failures.py
View File

@@ -192,7 +192,7 @@ class TestLinearizerFailures(unittest.TestCase):
[Opt(op=OptOps.GROUP, axis=0, amt=8), Opt(op=OptOps.UNROLL, axis=1, amt=4), Opt(op=OptOps.LOCAL, axis=0, amt=16), Opt(op=OptOps.UPCAST, axis=0, amt=0)],
]
for opts in all_failing_opts:
helper_test_lin(Linearizer(ast), opts, failed_platforms=["METAL", "GPU", "HSA", "CUDA"])
helper_test_lin(Linearizer(ast), opts, failed_platforms=[])
# COMPARE_ERROR from GPT2 kernel - just the first element off
# testing ast 41