From db2da22a047cc65f7f7f1e307606268ce163eed6 Mon Sep 17 00:00:00 2001 From: George Hotz Date: Sun, 30 Oct 2022 16:47:16 -0700 Subject: [PATCH] stop blowing up floats --- 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 568d8a4aee..fb25d5e9b5 100644 --- a/test/test_ops.py +++ b/test/test_ops.py @@ -6,7 +6,7 @@ import unittest from tinygrad.tensor import Tensor, Device FORWARD_ONLY = bool(int(os.getenv("FORWARD_ONLY", "0"))) -def helper_test_op(shps, torch_fxn, tinygrad_fxn, atol=1e-6, rtol=1e-3, grad_atol=1e-4, grad_rtol=1e-3, forward_only=False, vals=None, a=-0.5, b=20): +def helper_test_op(shps, torch_fxn, tinygrad_fxn, atol=1e-6, rtol=1e-3, grad_atol=1e-4, grad_rtol=1e-3, forward_only=False, vals=None, a=-0.5, b=3): torch.manual_seed(0) np.random.seed(0) if shps is None: