From 0d46530017d9dade97e37c02c35bbe2b87822876 Mon Sep 17 00:00:00 2001 From: Leandro Nunes Date: Fri, 15 Jan 2021 19:37:01 +0000 Subject: [PATCH] [TEST] Fix test_topi_batch_matmul_tensorcore.py:test_batch_matmul requirement * 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 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)