frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-21 05:34:18 +08:00
Code Issues 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
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)
@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: