You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tvm.apache.org by wu...@apache.org on 2021/03/04 22:04:26 UTC
[tvm] 05/11: test get_valid_counts on vulkan
This is an automated email from the ASF dual-hosted git repository.
wuwei pushed a commit to branch vk-i64
in repository https://gitbox.apache.org/repos/asf/tvm.git
commit acef292c42069abb94e09234dab701cc84271636
Author: Masahiro Masuda <ma...@gmail.com>
AuthorDate: Wed Mar 3 08:25:08 2021 +0900
test get_valid_counts on vulkan
---
tests/python/topi/python/test_topi_vision.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/tests/python/topi/python/test_topi_vision.py b/tests/python/topi/python/test_topi_vision.py
index 8393568..2fdf3cf 100644
--- a/tests/python/topi/python/test_topi_vision.py
+++ b/tests/python/topi/python/test_topi_vision.py
@@ -112,7 +112,7 @@ def verify_get_valid_counts(dshape, score_threshold, id_index, score_index):
tvm.testing.assert_allclose(tvm_out2.asnumpy(), np_out2, rtol=1e-3)
tvm.testing.assert_allclose(tvm_out3.asnumpy(), np_out3, rtol=1e-3)
- for device in ["llvm", "cuda", "opencl"]:
+ for device in ["llvm", "cuda", "opencl", "vulkan"]:
check_device(device)