mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
[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
This commit is contained in:
committed by
PyTorch MergeBot
parent
1600585219
commit
68dce23722
@ -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:
|
||||
|
Reference in New Issue
Block a user