github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-09 15:08:02 -05:00
Code Issues Packages Projects Releases Wiki Activity

combine pm_drop_and_clauses (#12660)

Browse Source 
* combine those

* wino kernels decreased
This commit is contained in:
Sieds Lykles
2025-10-14 10:09:41 +02:00
committed by GitHub
parent 84d4589ed4
commit e06cbfcb8a
2 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/unit/test_winograd.py
View File

@@ -42,7 +42,7 @@ class TestWinograd(unittest.TestCase):
out = Tensor.conv2d(x,w, padding=1) out = Tensor.conv2d(x,w, padding=1)
out.mean().backward() out.mean().backward()
backward_schedule = Tensor.schedule(x.grad, w.grad) backward_schedule = Tensor.schedule(x.grad, w.grad)
self.assertEqual(len(backward_schedule), 5) self.assertEqual(len(backward_schedule), 4)
def test_counters(self): def test_counters(self):
IC, OC, X, Y = 4,4,9,9 IC, OC, X, Y = 4,4,9,9

3
tinygrad/schedule/indexing.py
View File

@@ -128,8 +128,7 @@ def apply_movement_op(op:Ops, in_shape:tuple[sint,...], arg:tuple, rngs:tuple[UO
axes_out.append(combined_axes % s) axes_out.append(combined_axes % s)
combined_axes //= s combined_axes //= s
# this simplify is doing a lot of heavy lifting. this is the replacement for the reshape view merging code # this simplify is doing a lot of heavy lifting. this is the replacement for the reshape view merging code
rngs = graph_rewrite(graph_rewrite(UOp.sink(*axes_out[::-1]), symbolic+pm_simplify_valid, name="reshape"), rngs = graph_rewrite(UOp.sink(*axes_out[::-1]), symbolic+pm_simplify_valid+pm_drop_and_clauses, name="reshape").src
pm_drop_and_clauses, name="reshape drop ands").src
case _: raise RuntimeError(f"{op} is not a MovementOp") case _: raise RuntimeError(f"{op} is not a MovementOp")
return rngs return rngs