[TOPI] Specify non-zero absolute tolerance in tests (apache#1925)

tqchen · Oct 21, 2018 · 9ab4128 · 9ab4128
1 parent 6500bb3
commit 9ab4128
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/tests/python/integration/test_benchmark_gemm.py b/tests/python/integration/test_benchmark_gemm.py
@@ -94,7 +94,7 @@ def verify(s, check_correctness=True):
                                                    env.BATCH,
                                                    env.BLOCK_OUT)
             if check_correctness:
-                np.testing.assert_allclose(res_unpack, res_ref)
+                tvm.testing.assert_allclose(res_unpack, res_ref)
             return cost
 
         def run_schedule(load_inp,

diff --git a/tests/python/integration/test_benchmark_topi_conv2d.py b/tests/python/integration/test_benchmark_topi_conv2d.py
@@ -87,7 +87,7 @@ def verify(s, check_correctness):
                 padding = wl.hpad
                 res_ref = res_ref >> 8
                 res_ref = np.clip(res_ref, 0, 127).astype("int8")
-                np.testing.assert_allclose(res_unpack, res_ref)
+                tvm.testing.assert_allclose(res_unpack, res_ref)
             return cost
 
         def conv_normal(print_ir):
@@ -219,7 +219,7 @@ def verify(s, check_correctness):
                 res_ref = res_ref >> 8
                 res_ref += bias_orig.reshape(wl.out_filter, 1, 1)
                 res_ref = np.clip(res_ref, 0, 127).astype("int8")
-                np.testing.assert_allclose(res_unpack, res_ref)
+                tvm.testing.assert_allclose(res_unpack, res_ref)
             return cost
 
         def conv_normal(print_ir):

diff --git a/tutorials/convolution_opt.py b/tutorials/convolution_opt.py
@@ -413,7 +413,7 @@
                            env.BLOCK_OUT,
                            fout_height,
                            fout_width)).transpose((0, 2, 4, 5, 1, 3))
-np.testing.assert_allclose(res_ref, res_nd.asnumpy())
+tvm.testing.assert_allclose(res_ref, res_nd.asnumpy())
 print("Successful 2D convolution test!")
 
 ######################################################################