From fee29cdbf67d10abb09f8a6f1e4894defc59c80f Mon Sep 17 00:00:00 2001 From: Masahiro Masuda Date: Tue, 9 Mar 2021 05:15:43 +0900 Subject: [PATCH] add nvptx to sort test too --- tests/python/topi/python/test_topi_sort.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/python/topi/python/test_topi_sort.py b/tests/python/topi/python/test_topi_sort.py index 0879a0aa9dfe..85a35488ab22 100644 --- a/tests/python/topi/python/test_topi_sort.py +++ b/tests/python/topi/python/test_topi_sort.py @@ -75,7 +75,7 @@ def check_device(device): f(tvm_data, tvm_out) tvm.testing.assert_allclose(tvm_out.asnumpy(), np_sort, rtol=1e0) - for device in ["llvm", "cuda", "opencl", "vulkan"]: + for device in ["llvm", "cuda", "opencl", "vulkan", "nvptx"]: check_device(device) @@ -115,7 +115,7 @@ def check_device(device): f(tvm_data, tvm_out) tvm.testing.assert_allclose(tvm_out.asnumpy(), np_indices.astype(data_dtype), rtol=1e0) - for device in ["llvm", "cuda", "opencl", "vulkan"]: + for device in ["llvm", "cuda", "opencl", "vulkan", "nvptx"]: check_device(device) @@ -167,7 +167,7 @@ def check_device(device): else: tvm.testing.assert_allclose(tvm_res[0].asnumpy(), np_indices) - for device in ["llvm", "cuda", "opencl", "vulkan"]: + for device in ["llvm", "cuda", "opencl", "vulkan", "nvptx"]: check_device(device)