[TEST] Fix test_topi_batch_matmul_tensorcore.py:test_batch_matmul req…

…uirement (apache#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 apache#7277
masahi · Jan 18, 2021 · d1b9d59 · d1b9d59
1 parent 6137171
commit d1b9d59
Showing 1 changed file with 1 addition and 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
@@ -63,7 +63,7 @@ def check_device(device):
     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)