From 339a35b081d3c87a15dd596356fa6d617d23e142 Mon Sep 17 00:00:00 2001 From: George Hotz Date: Fri, 30 Oct 2020 08:32:16 -0700 Subject: [PATCH] div needs help --- test/test_ops.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/test_ops.py b/test/test_ops.py index 06195fa8af..e110c76173 100644 --- a/test/test_ops.py +++ b/test/test_ops.py @@ -39,7 +39,7 @@ class TestOps(unittest.TestCase): helper_test_op([(45,65), (45,65)], lambda x,y: x*y, Tensor.mul) def test_div(self): # TODO: why does this need more tolerance? - helper_test_op([(45,65), (45,65)], lambda x,y: x/y, Tensor.div, atol=5e-5) + helper_test_op([(45,65), (45,65)], lambda x,y: x/y, Tensor.div, atol=5e-5, grad_atol=1e-5) def test_pow(self): helper_test_op([(45,65), (45,65)], lambda x,y: x**y, Tensor.pow) def test_sqrt(self):