Fix test

apache · Jul 10, 2018 · 39cc81c · 39cc81c
1 parent 473f603
commit 39cc81c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/topi/tests/python_cpp/test_topi_transform.py b/topi/tests/python_cpp/test_topi_transform.py
@@ -232,7 +232,7 @@ def check_device(device):
             tvm.nd.array(y), tvm_out)
         np.testing.assert_allclose(tvm_out.asnumpy(), np_out)
 
-    for device in ["llvm", "nvptx", "cuda", "opencl", "metal", "rocm"]:
+    for device in ["llvm", "cuda", "opencl", "metal", "rocm"]:
         check_device(device)
 
 def verify_concatenate_split(shapes, axis, indices_or_sections):