OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 00:21:07 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

[ROCm] Skip test_jit_cudnn_extension on ROCm (#105805)

Browse Source 
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
This commit is contained in:
Pruthvi Madugundu 2023-07-24 14:19:53 +00:00 committed by PyTorch MergeBot
parent 1600585219
commit 68dce23722
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
test/test_cpp_extensions_jit.py
View File

@ -241,6 +241,7 @@ class TestCppExtensionJIT(common.TestCase):
self._run_jit_cuda_archflags(flags, expected) self._run_jit_cuda_archflags(flags, expected)
@unittest.skipIf(not TEST_CUDNN, "CuDNN not found") @unittest.skipIf(not TEST_CUDNN, "CuDNN not found")
@unittest.skipIf(TEST_ROCM, "Not supported on ROCm")
def test_jit_cudnn_extension(self): def test_jit_cudnn_extension(self):
# implementation of CuDNN ReLU # implementation of CuDNN ReLU
if IS_WINDOWS: if IS_WINDOWS: