[tvm] branch main updated: [TEST] Fix test_topi_batch_matmul_tensorcore.py:test_batch_matmul requirement (#7294)

[tqchen]

This is an automated email from the ASF dual-hosted git repository.

tqchen pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git


The following commit(s) were added to refs/heads/main by this push:
     new 3f15d06  [TEST] Fix 
test_topi_batch_matmul_tensorcore.py:test_batch_matmul requirement (#7294)
3f15d06 is described below

commit 3f15d062e335b83c5170de5ee9ae19b90dcb681c
Author: Leandro Nunes <leandro...@gmail.com>
AuthorDate: Sat Jan 16 14:48:08 2021 +0000

    [TEST] Fix test_topi_batch_matmul_tensorcore.py:test_batch_matmul 
requirement (#7294)
    
    * this test current sets a requirement to "uses_gpu", which
       causes it to fail in cpu-only machine
     * this patch changes it to be "requires_tensorcore", as per discussion
       on issue #7277
---
 tests/python/topi/python/test_topi_batch_matmul_tensorcore.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/python/topi/python/test_topi_batch_matmul_tensorcore.py 
b/tests/python/topi/python/test_topi_batch_matmul_tensorcore.py
index 60f4bef3..77df5be 100644
--- a/tests/python/topi/python/test_topi_batch_matmul_tensorcore.py
+++ b/tests/python/topi/python/test_topi_batch_matmul_tensorcore.py
@@ -63,7 +63,7 @@ def verify_batch_matmul(x_batch, y_batch, M, N, K):
     check_device("cuda")
 
 
-@tvm.testing.uses_gpu
+@tvm.testing.requires_tensorcore
 def test_batch_matmul():
     verify_batch_matmul(1, 1, 16, 16, 32)
     verify_batch_matmul(5, 5, 16, 16, 32)