From 39cc81c006c1d95fad0c888441d94fb664b9362c Mon Sep 17 00:00:00 2001 From: Wang Date: Tue, 10 Jul 2018 14:50:06 -0700 Subject: [PATCH] Fix test --- topi/tests/python_cpp/test_topi_transform.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/topi/tests/python_cpp/test_topi_transform.py b/topi/tests/python_cpp/test_topi_transform.py index 0471352fc2a52..2a12577373267 100644 --- 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):