Skip to content

Commit

Permalink
add nvptx to sort test too
Browse files Browse the repository at this point in the history
  • Loading branch information
masahi committed Mar 8, 2021
1 parent 2030786 commit fee29cd
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions tests/python/topi/python/test_topi_sort.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)


Expand Down Expand Up @@ -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)


Expand Down Expand Up @@ -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)


Expand Down

0 comments on commit fee29cd

Please sign in to comment.