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

avgpool2d is a second class op

Browse Source
This commit is contained in:
George Hotz
2020-12-28 10:41:59 -05:00
parent 593233b668
commit fafece9db7
5 changed files with 15 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/test_ops.py
View File

@@ -169,7 +169,7 @@ class TestOps(unittest.TestCase):
with self.subTest(kernel_size=ksz):
helper_test_op([shape],
lambda x: torch.nn.functional.avg_pool2d(x, kernel_size=ksz),
lambda x: Tensor.avg_pool2d(x, kernel_size=ksz), device=self.device)
lambda x: Tensor.avg_pool2d(x, kernel_size=ksz), device=self.device, rtol=1e-5)
@unittest.skipUnless(GPU, "Requires GPU")
class TestOpsGPU(TestOps):