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

use the PatternMatcher to validate UOps type [run_process_replay] (#6583)

Browse Source 
* use the PatternMatcher to validate UOps type [run_process_replay]

* type check tests pass

* DEFINE_VAR

* fix precommit

* fix tests

* ptx

* type check tests pass

* ptx test

* int64

* ptx barrier

* delete old stuff
This commit is contained in:
George Hotz
2024-09-19 09:59:06 +08:00
committed by GitHub
parent d01e011a8c
commit fa0f678d5a
2 changed files with 83 additions and 264 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
test/test_renderer_failures.py
View File

@@ -50,6 +50,7 @@ class TestPTXFailures(unittest.TestCase):
ret = _test_uop_result([], uops, local_size=[4, 1, 1])[0]
np.testing.assert_equal(ret, [0, 1, 1, 1])
@unittest.skip("not still valid?")
def test_gated_store_with_if(self):
a = UOp(UOps.DEFINE_GLOBAL, PtrDType(dtypes.int), (), 0)
gate_alu = (lidx0:=UOp(UOps.SPECIAL, dtypes.int, (), ('lidx0', 4))).ne(0)