From 745316493ca6a2268c786ecffd12e6df1cbfda0f Mon Sep 17 00:00:00 2001 From: George Hotz Date: Sun, 10 Nov 2024 18:36:42 +0800 Subject: [PATCH] hotfix: add test_simple_conv2d_bias --- test/test_ops.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/test/test_ops.py b/test/test_ops.py index 62e69992e3..ca1a95af62 100644 --- a/test/test_ops.py +++ b/test/test_ops.py @@ -1483,6 +1483,11 @@ class TestOps(unittest.TestCase): lambda x,w: torch.nn.functional.conv2d(x,w).relu(), lambda x,w: Tensor.conv2d(x,w).relu(), grad_rtol=1e-5) + def test_simple_conv2d_bias(self): + helper_test_op([(1,4,9,9), (4,4,3,3), (4,)], + lambda x,w,b: torch.nn.functional.conv2d(x,w,b).relu(), + lambda x,w,b: Tensor.conv2d(x,w,b).relu(), grad_rtol=1e-5) + @unittest.skipIf(IMAGE>0, "no conv3d on images") def test_simple_conv3d(self): helper_test_op([(1,4,9,9,9), (4,4,3,3,3)],