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

move reduce to lowerer [pr] (#7462)

Browse Source 
* move reduce to lowerer [pr]

* simpler
This commit is contained in:
George Hotz
2024-11-01 15:39:20 +07:00
committed by GitHub
parent 2cfca230b5
commit a7ba3d2d91
4 changed files with 21 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/unit/test_graph_rewrite.py
View File

@@ -51,6 +51,7 @@ class TestArithmeticSimplifications(unittest.TestCase):
class TestFoldingAndReduction(unittest.TestCase):
@unittest.skip("reduce is removed now")
def test_full_graph_rewrite_constant_reduction_folding(self):
const1 = UOp.const(dtypes.int32, 5)
const2 = UOp.const(dtypes.int32, 10)
@@ -59,6 +60,7 @@ class TestFoldingAndReduction(unittest.TestCase):
expected_sum = 5 + 10 + 20
self.assertEqual(optimized_sink.arg, expected_sum)
@unittest.skip("reduce is removed now")
def test_full_graph_rewrite_reduction_with_unused_range(self):
const1 = UOp.const(dtypes.int32, 15)
const2 = UOp.const(dtypes.int32, 25)