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 60f4bef3a855..77df5be0a491 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 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)