From 68dce23722428f3035a61cf16a006fb148bf6c43 Mon Sep 17 00:00:00 2001
From: Pruthvi Madugundu <pruthvigithub@gmail.com>
Date: Mon, 24 Jul 2023 14:19:53 +0000
Subject: [PATCH] [ROCm] Skip test_jit_cudnn_extension on ROCm (#105805)

follow up https://github.com/pytorch/pytorch/pull/105594

Pull Request resolved: https://github.com/pytorch/pytorch/pull/105805
Approved by: https://github.com/ezyang
---
 test/test_cpp_extensions_jit.py | 1 +
 1 file changed, 1 insertion(+)

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: