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

remove NEG from handwritten ast in tests (#6234)

Browse Source 
* remove NEG from handwritten ast in tests

* test_linearizer_failures
This commit is contained in:
chenyu
2024-08-22 09:06:59 -04:00
committed by GitHub
parent 1c5ef5b793
commit 3fc8203475
7 changed files with 75 additions and 107 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/test_pattern_matcher.py
View File

@@ -132,7 +132,7 @@ class TestPatternMatcher(TestUOps):
c1 = UOp(UOps.CONST, dtypes.float, arg=1.0)
c2 = UOp(UOps.CONST, dtypes.float, arg=2.0)
c3 = UOp(UOps.CONST, dtypes.float, arg=3.0)
c4 = UOp(UOps.ALU, dtypes.float, (c1,), UnaryOps.NEG)
c4 = UOp(UOps.ALU, dtypes.float, (c1,), UnaryOps.EXP2)
c5 = UOp(UOps.ALU, dtypes.float, (c1,c2), BinaryOps.ADD)
c6 = UOp(UOps.ALU, dtypes.float, (c1,c2,c3), TernaryOps.MULACC)
self.assertEqual(matcher.rewrite(c4), None)