diff --git a/test/test_cpp_extensions_jit.py b/test/test_cpp_extensions_jit.py index faf54ac5c70..deba17fa26e 100644 --- a/test/test_cpp_extensions_jit.py +++ b/test/test_cpp_extensions_jit.py @@ -241,6 +241,7 @@ class TestCppExtensionJIT(common.TestCase): self._run_jit_cuda_archflags(flags, expected) @unittest.skipIf(not TEST_CUDNN, "CuDNN not found") + @unittest.skipIf(TEST_ROCM, "Not supported on ROCm") def test_jit_cudnn_extension(self): # implementation of CuDNN ReLU if IS_WINDOWS: