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:28 UTC

[tvm] 07/11: pytest fix

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 62889b04edb5b97b5269d2f7d5715de56319d846
Author: Masahiro Masuda <ma...@gmail.com>
AuthorDate: Wed Mar 3 18:14:23 2021 +0900

    pytest fix
---
 tests/python/topi/python/test_topi_cumsum.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/python/topi/python/test_topi_cumsum.py b/tests/python/topi/python/test_topi_cumsum.py
index bf962d9..6b99239 100644
--- a/tests/python/topi/python/test_topi_cumsum.py
+++ b/tests/python/topi/python/test_topi_cumsum.py
@@ -41,7 +41,7 @@ def test_cumsum(ctx, target):
     check_cumsum(np.cumsum(data, dtype=np.int32), data)
     check_cumsum(np.cumsum(data), data, dtype="int64")
 
-    if str(target.kind) != "vulkan":
+    if target != "vulkan":
         # TODO(masahi): Support bool tensor in SPIRV codegen
         data = np.random.rand(10) > 0.5
         check_cumsum(np.cumsum(data, dtype=np.int32), data, dtype="int32")