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

Revert "threefry_2x32 (#2601)" (#3784)

Browse Source 
This reverts commit db3de54bc4.
This commit is contained in:
George Hotz
2024-03-17 10:27:20 -07:00
committed by GitHub
parent db3de54bc4
commit 311cf2b7d3
13 changed files with 56 additions and 92 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
test/test_schedule.py
View File

@@ -181,7 +181,7 @@ class TestSchedule(unittest.TestCase):
# run
img = Tensor.rand(2,3,64,64)
out = c1(img).elu()
check_schedule(out, 1, [c1.weight, c1.bias, img])
check_schedule(out, 1, [c1.weight, c1.bias])
def test_two_sum(self):
img = Tensor.empty(64,64)
@@ -336,7 +336,7 @@ class TestSchedule(unittest.TestCase):
out = bn1(conv1(x)).relu()
out = bn2(conv2(out))
out = (out + x).relu()
check_schedule(out, 2, [conv1.weight, conv2.weight])
check_schedule(out, 4)
def test_contiguous_while_contiguous(self):
x = Tensor.empty(1, 64, 32, 32)