From bbb46456b5a7d05b11d95f91c53842be9b29e13c Mon Sep 17 00:00:00 2001 From: Mehrdad Hessar Date: Tue, 10 Aug 2021 15:51:14 -0700 Subject: [PATCH] increase atol for float32 --- tests/python/topi/python/test_topi_conv2d_nchw.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/python/topi/python/test_topi_conv2d_nchw.py b/tests/python/topi/python/test_topi_conv2d_nchw.py index 7f982d56ae06..5cc8bb0a1de7 100644 --- a/tests/python/topi/python/test_topi_conv2d_nchw.py +++ b/tests/python/topi/python/test_topi_conv2d_nchw.py @@ -142,7 +142,7 @@ def test_conv2d_nchw( if "int" in dtype: tol = {"atol": 0, "rtol": 0} elif dtype == "float32": - tol = {"rtol": 1e-4, "atol": 1e-5} + tol = {"rtol": 1e-4, "atol": 1e-4} elif dtype == "float16": # A summation in float16 with a single accumulator very # quickly runs into large rounding errors. At some point,