Add test guard

apache · Apr 7, 2022 · 73f352f · 73f352f
1 parent 27d058e
commit 73f352f
Showing 1 changed file with 1 addition and 0 deletions.
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",
     [