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

tests from lowerer branch (#5339)

Browse Source 
* tests from lowerer branch

* Update test_image_dtype.py

* Update test_image_dtype.py

* Update test_image_dtype.py
This commit is contained in:
George Hotz
2024-07-08 21:23:19 -07:00
committed by GitHub
parent 4ceab5d2b1
commit c13da83f12
5 changed files with 24 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
test/test_conv.py
View File

@@ -42,6 +42,13 @@ class TestConv(unittest.TestCase):
print(ret.numpy())
def test_two_binops_no_rerun_small(self):
Tensor.no_grad = True
x = Tensor.rand(1,1,32,32)
w = Tensor.rand(1,1,3,3)
out = x.conv2d(w, padding=(1,1))
np.testing.assert_allclose(out.relu().numpy(), np.maximum(out.numpy(), 0))
def test_two_binops_no_rerun(self):
Tensor.no_grad = True
x = Tensor.randn(1,12,128,256)