Revert "[RELAND] Add UTs for accelerator device-agnostic runtime APIs (#133572)"

pytorchmergebot · pytorchmergebot · commit 1b3f8b758967 · 2024-12-11T21:47:18.000Z
This reverts commit 2091194. Reverted #133572 on behalf of https://github.com/huydhn due to Sorry for reverting your change but the new test is still very flaky on MacOS even when it does not segfault anymore ([comment](#133572 (comment)))
diff --git a/test/test_accelerator.py b/test/test_accelerator.py
diff --git a/test/test_cuda.py b/test/test_cuda.py
@@ -725,14 +725,6 @@ def test_generic_stream_event(self):
         self.assertTrue(issubclass(type(cuda_event), torch.Event))
         self.assertTrue(torch.Event in type(cuda_event).mro())
 
-    def test_stream_compatibility(self):
-        s1 = torch.cuda.Stream()
-        s2 = torch.cuda.Stream()
-        torch.accelerator.set_stream(s1)
-        self.assertEqual(torch.accelerator.current_stream().stream_id, s1.stream_id)
-        torch.accelerator.set_stream(s2)
-        self.assertEqual(torch.accelerator.current_stream().stream_id, s2.stream_id)
-
     def test_record_stream(self):
         cycles_per_ms = get_cycles_per_ms()
 
diff --git a/test/test_xpu.py b/test/test_xpu.py
@@ -299,14 +299,6 @@ def test_generic_stream_event(self):
         self.assertTrue(issubclass(type(xpu_event), torch.Event))
         self.assertTrue(torch.Event in type(xpu_event).mro())
 
-    def test_stream_compatibility(self):
-        s1 = torch.xpu.Stream()
-        s2 = torch.xpu.Stream()
-        torch.accelerator.set_stream(s1)
-        self.assertEqual(torch.accelerator.current_stream().stream_id, s1.stream_id)
-        torch.accelerator.set_stream(s2)
-        self.assertEqual(torch.accelerator.current_stream().stream_id, s2.stream_id)
-
     def test_generator(self):
         torch.manual_seed(2024)
         g_state0 = torch.xpu.get_rng_state()
