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

test: uop and lazyop have the same compare (#6053)

Browse Source 
* test: uop and lazyop have the same compare

* typings

* self.assert_equiv_uops -> assertEqual

* hash dtype

* test nop too

* TestPatternMatcher never used this compare anyway

* nop eq and ne tests
This commit is contained in:
qazal
2024-08-13 05:33:19 +08:00
committed by GitHub
parent 8c501272f3
commit dccca7f227
5 changed files with 86 additions and 66 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
test/test_pattern_matcher.py
View File

@@ -1,11 +1,10 @@
import unittest, itertools
from test.helpers import TestUOps
from tinygrad.dtype import dtypes
from tinygrad.ops import BinaryOps, TernaryOps, ReduceOps, UnaryOps # noqa: F401
from tinygrad.codegen.uops import UOps, UOp, PatternMatcher, UPat
from tinygrad.codegen.uopgraph import constant_folder
class TestPatternMatcher(TestUOps):
class TestPatternMatcher(unittest.TestCase):
def test_simple_match(self):
matcher = PatternMatcher([(UPat(UOps.CONST, name="x", dtype=dtypes.float), lambda x: x)])
c1 = UOp(UOps.CONST, dtypes.float, arg=1.0)