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

UOp simple mul add div fold (#5726)

Browse Source
This commit is contained in:
chenyu
2024-07-25 22:00:30 -04:00
committed by GitHub
parent 78f75aa80d
commit 2cc55a3095
2 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
test/unit/test_uop_symbolic.py
View File

@@ -212,11 +212,9 @@ class TestSymbolic(unittest.TestCase):
# This is mod reduction
self.helper_test_variable((1+Variable("a",1,2))%2, 0, 1, {"(-1+a)", "(a+(-1))"})
@unittest.expectedFailure
def test_sum_div_const(self):
self.helper_test_variable(Node.sum([Variable("a", 0, 7)*4, NumNode(3)]) // 4, 0, 7, "a")
@unittest.expectedFailure
def test_sum_div_const_big(self):
self.helper_test_variable(Node.sum([Variable("a", 0, 7)*4, NumNode(3)]) // 16, 0, 1, "(a//4)")
@@ -311,7 +309,6 @@ class TestSymbolic(unittest.TestCase):
def test_mul_div_factor_div(self):
self.helper_test_variable((Variable("a", 0, 10)*4)//8, 0, 5, "(a//2)")
@unittest.expectedFailure
def test_sum_div_partial_remove(self):
self.helper_test_variable(Node.sum([Variable("idx0", 0, 127)*4, Variable("idx2", 0, 3)])//4, 0, 127, "idx0")