From ff6655defb5f95ed6ac5e77f929c6fa8ea0f1939 Mon Sep 17 00:00:00 2001 From: Jeff Daily Date: Fri, 8 Jul 2022 21:29:29 +0000 Subject: [PATCH] [ROCm] unskip external streams tests (#80922) These two tests are passing for ROCm 5.1.1 and 5.2. Pull Request resolved: https://github.com/pytorch/pytorch/pull/80922 Approved by: https://github.com/cpuhrsch --- test/test_cuda.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/test/test_cuda.py b/test/test_cuda.py index a1d1622299c..9db58398881 100644 --- a/test/test_cuda.py +++ b/test/test_cuda.py @@ -1351,7 +1351,6 @@ class TestCuda(TestCase): out = cudart.cudaStreamDestroy(stream.value) self.assertEqual(out, 0) - @skipIfRocm def test_external_streams(self): device = torch.cuda.device(0) with self._get_external_stream(device) as stream_v: @@ -1359,7 +1358,6 @@ class TestCuda(TestCase): self.assertEqual(stream_v, ext_stream.cuda_stream) self.assertEqual(ext_stream.device.index, device.idx) - @skipIfRocm @unittest.skipIf(not TEST_MULTIGPU, "detected only one GPU") def test_external_streams_multi_device(self): device = torch.cuda.device(1)