Add test guard

mbaret · mbaret · commit 057b91867da6 · 2022-04-06T19:47:46.000+01:00
diff --git a/tests/python/contrib/test_cudnn.py b/tests/python/contrib/test_cudnn.py
@@ -488,6 +488,7 @@ def _verify_cudnn_relay(expr):
     )
 
 
+@tvm.testing.requires_cuda
 @pytest.mark.parametrize(
     "shape,axis",
     [
