From b9a6c091075fdca4ce37372efaf0bdef1c145776 Mon Sep 17 00:00:00 2001
From: Xingjian Shi <xshiab@ust.hk>
Date: Tue, 16 Jan 2018 10:30:14 -0800
Subject: [PATCH] fix (#788)
---
topi/tests/python/test_topi_reduce.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/topi/tests/python/test_topi_reduce.py b/topi/tests/python/test_topi_reduce.py
index c8d95df25..1e40770f5 100644
--- 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)
--
GitLab