diff --git a/tests/python/contrib/test_cudnn.py b/tests/python/contrib/test_cudnn.py index cb88b62021e4..45ca7c91717d 100644 --- 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", [