Add missing annotation for requires_gpu in test_topi_dense.py Require…

…s GPU (apache#8387)
zxy844288792 · Jan 13, 2022 · 58096e1 · 58096e1
1 parent 99071af
commit 58096e1
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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,