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

multiple locals + get_kernel_modifier + fix valid (#5739)

Browse Source 
* multiple locals + get_kernel_modifier + fix valid

* fix test pattern matcher
This commit is contained in:
George Hotz
2024-07-26 15:10:10 -07:00
committed by GitHub
parent f6c0e17a2c
commit c50e374bb6
5 changed files with 19 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
test/test_pattern_matcher.py
View File

@@ -166,6 +166,11 @@ class TestPatternMatcher(TestUOps):
dtypes._float2 = dtypes.float.vec(2)
dtypes._float4 = dtypes.float.vec(4)
dtypes._float8 = dtypes.float.vec(8)
dtypes._float16 = dtypes.float.vec(16)
dtypes._half2 = dtypes.half.vec(2)
dtypes._half4 = dtypes.half.vec(4)
dtypes._half8 = dtypes.half.vec(8)
dtypes._half16 = dtypes.half.vec(16)
upat = UPat(UOps.CONST, name="x", dtype=dtypes.float)
assert str(upat) == str(eval(str(upat)))
evpat:UPat = eval(repr(UPat(src = [UPat(name='a'), UPat(name='b')])))