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

Add variations for some ADD patterns (#8393)

Browse Source 
* Add variations for some ADD patterns

* Add test and remove redundant rule
This commit is contained in:
Sieds Lykles
2024-12-26 01:49:39 +01:00
committed by GitHub
parent a562ee2c6e
commit 6bb54eb532
2 changed files with 10 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
test/unit/test_uop_symbolic.py
View File

@@ -90,11 +90,16 @@ class TestSymbolic(unittest.TestCase):
def test_factorize(self):
a = Variable("a", 0, 8)
b = Variable("b", 0, 8)
self.helper_test_variable(a*2+a*3, 0, 8*5, "(a*5)")
self.helper_test_variable(b+a*2+a*3, 0, 8*6, "(b+(a*5))")
def test_factorize_no_mul(self):
a = Variable("a", 0, 8)
b = Variable("b", 0, 8)
self.helper_test_variable(a+a*3, 0, 8*4, "(a*4)")
self.helper_test_variable((a+b)+a*3, 0, 8*5, "(b+(a*4))")
self.helper_test_variable((a*3+b)+b*3, 0, 8*7, "((a*3)+(b*4))")
def test_neg(self):
self.helper_test_variable(-Variable("a", 0, 8), -8, 0, "(a*-1)")
@@ -107,7 +112,9 @@ class TestSymbolic(unittest.TestCase):
def test_add_self(self):
a = Variable("a", 0, 8)
b = Variable("b", 0, 8)
self.helper_test_variable(a+a, 0, 16, "(a*2)")
self.helper_test_variable((a+b)+b, 0, 24, "(a+(b*2))")
def test_sub_self(self):
a = Variable("a", 0, 8)