allow tests to run without cudnn
@@ -28,14 +28,11 @@ class TestCheckVersion(unittest.TestCase):
assert isinstance(build.get_hip_version(), int)
assert isinstance(build.get_hip_version(True), str)
- @pytest.mark.gpu
- @pytest.mark.cudnn
@pytest.mark.xfail(test_hip,
reason='ROCm/HIP DNN support is not ready')
def test_check_cudnn_version(self):
with self.assertRaises(RuntimeError):
build.get_cudnn_version()
- assert build.check_cudnn_version(
- self.compiler, self.settings)
+ if build.check_cudnn_version(self.compiler, self.settings):
assert isinstance(build.get_cudnn_version(), int)
assert isinstance(build.get_cudnn_version(True), str)
