From b10bc82991fab2efaed380b567a72a1c387f6da3 Mon Sep 17 00:00:00 2001
From: Bo Li <110066325+BLOrange-AMD@users.noreply.github.com>
Date: Fri, 15 Nov 2024 15:12:41 -0600
Subject: [PATCH] Updated test_fully_shard_compile.py to skip unit tests

---
 .../_composable/fully_shard/test_fully_shard_compile.py        | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/test/distributed/_composable/fully_shard/test_fully_shard_compile.py b/test/distributed/_composable/fully_shard/test_fully_shard_compile.py
index 03c503f732a19..3913c4eb9b8b8 100644
--- a/test/distributed/_composable/fully_shard/test_fully_shard_compile.py
+++ b/test/distributed/_composable/fully_shard/test_fully_shard_compile.py
@@ -17,7 +17,7 @@
     FSDPTest,
     TransformerWithSharedParams,
 )
-from torch.testing._internal.common_utils import run_tests, TEST_WITH_DEV_DBG_ASAN
+from torch.testing._internal.common_utils import run_tests, TEST_WITH_DEV_DBG_ASAN, skipIfRocm
 from torch.utils._triton import has_triton
 
 if not dist.is_available():
@@ -39,6 +39,7 @@ def world_size(self) -> int:
 
     @unittest.skipIf(not has_triton(), "Inductor+gpu needs triton and recent GPU arch")
     @skip_if_lt_x_gpu(2)
+    @skipIfRocm # temp skip
     def test_compile(self):
         self.run_subtests(
             {