From 9ca98ca9f20016d17ec2925ad677e5706352647e Mon Sep 17 00:00:00 2001 From: Ramana Radhakrishnan Date: Thu, 1 Jul 2021 16:54:14 +0100 Subject: [PATCH] Add missing annotation for requires_gpu in test_topi_dense.py Requires GPU --- tests/python/topi/python/test_topi_dense.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/python/topi/python/test_topi_dense.py b/tests/python/topi/python/test_topi_dense.py index 235a09400387..964c1621fa47 100644 --- a/tests/python/topi/python/test_topi_dense.py +++ b/tests/python/topi/python/test_topi_dense.py @@ -135,6 +135,7 @@ def test_dense( @pytest.mark.parametrize("target,in_dtype,out_dtype", [("cuda", "int8", "int32")]) +@tvm.testing.requires_gpu def test_dense_cuda_int8( target, dev,