github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-07 03:00:26 -04:00
Code Issues Packages Projects Releases Wiki Activity

fix cifar training in RANGEIFY (#12355)

Browse Source 
* fix cifar training in RANGEIFY

* even more wino fuse

* bugfix

* test to show issue
This commit is contained in:
George Hotz
2025-09-30 15:59:19 +08:00
committed by GitHub
parent 4ff7f20b9d
commit 7129419500
5 changed files with 43 additions and 6 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.mean().backward()
backward_schedule = Tensor.schedule(x.grad, w.grad)
self.assertEqual(len(backward_schedule), 4 if RANGEIFY else 9)
self.assertEqual(len(backward_schedule), 3 if RANGEIFY else 9)
def test_counters(self):
IC, OC, X, Y = 4,4,9,9