From 73f352ff035298451239dc4ccd21e1d732753a7c Mon Sep 17 00:00:00 2001
From: Matthew Barrett <mbarrett@octoml.ai>
Date: Wed, 6 Apr 2022 19:47:46 +0100
Subject: [PATCH] Add test guard

---
 tests/python/contrib/test_cudnn.py | 1 +
 1 file changed, 1 insertion(+)

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",
     [