[UnitTest] Updated tolerances to avoid flaky unit test. (#8723)

* [UnitTest] Updated tolerances to avoid flaky unit test. The result was correct, but the atol was just small enough to trigger a CI error for a value that was close to zero in an unrelated PR at #8670. https://ci.tlcpack.ai/blue/organizations/jenkins/tvm/detail/PR-8670/16/pipeline/#step-236-log-1703 * Also updated 32-bit version of test_conv2d_nchw
apache · Aug 12, 2021 · 8843153 · 8843153
1 parent 3e37bb5
commit 8843153
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/python/relay/test_auto_scheduler_layout_rewrite_networks.py b/tests/python/relay/test_auto_scheduler_layout_rewrite_networks.py
@@ -179,7 +179,7 @@ def get_output(data, lib):
         actual_output = get_output(data, lib)
         expected_output = get_output(data, lib2)
 
-        tvm.testing.assert_allclose(actual_output, expected_output, rtol=1e-4, atol=1e-4)
+        tvm.testing.assert_allclose(actual_output, expected_output, rtol=1e-4, atol=2e-4)
 
 
 def test_conv2d():

diff --git 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-4}
+            tol = {"rtol": 1e-4, "atol": 2e-4}
         elif dtype == "float16":
             # A summation in float16 with a single accumulator very
             # quickly runs into large rounding errors.  At some point,