fix (#788)

b9a6c091 · Xingjian Shi · Tianqi Chen · 16694815 · b9a6c091
Commit b9a6c091 authored Jan 16, 2018 by Xingjian Shi Committed by Tianqi Chen Jan 16, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

topi/tests/python/test_topi_reduce.py
+2 -2

No files found.
--- a/topi/tests/python/test_topi_reduce.py
+++ b/topi/tests/python/test_topi_reduce.py
@@ -88,8 +88,8 @@ def verify_reduce_map_ele(in_shape, axis, keepdims, type="sum"):
                np.testing.assert_allclose(out_tvm_val, in_npy_map.max(axis=axis), 1E-3, 1E-3)
            elif type == "argmin":
                np.testing.assert_allclose(out_tvm_val, in_npy_map.min(axis=axis), 1E-3, 1E-3)
-
-        np.testing.assert_allclose(out_tvm.asnumpy(), out_npy, 1E-3, 1E-3)
+        else:
+            np.testing.assert_allclose(out_tvm.asnumpy(), out_npy, 1E-3, 1E-3)
    for device in ["cuda", "opencl", "metal", "llvm", "rocm"]:
        check_device(device)