Only run that test for arm-cpu

apache · Aug 13, 2019 · 877b0ee · 877b0ee
1 parent 14daf78
commit 877b0ee
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/topi/tests/python/test_topi_math.py b/topi/tests/python/test_topi_math.py
@@ -68,8 +68,7 @@ def check_device(device):
             foo(a, b)
             tvm.testing.assert_allclose(b.asnumpy(), b_np, rtol=1e-5, atol=1e-5)
 
-        for device in get_all_backend():
-            check_device(device)
+        check_device('llvm -device=arm-cpu')
 
     test_apply(topi.floor, "floor", np.floor, -100, 100)
     test_apply(topi.ceil, "ceil", np.ceil, -100, 100)