diff --git a/test/dynamo/test_structured_trace.py b/test/dynamo/test_structured_trace.py
index 85d52f32cc5a1f..e65a0120cc8889 100644
--- a/test/dynamo/test_structured_trace.py
+++ b/test/dynamo/test_structured_trace.py
@@ -19,7 +19,7 @@
 from torch._inductor.test_case import TestCase
 from torch._logging._internal import TorchLogsFormatter
 from torch.nn.parallel import DistributedDataParallel as DDP
-from torch.testing._internal.common_utils import find_free_port
+from torch.testing._internal.common_utils import find_free_port, skipIfRocm
 from torch.testing._internal.inductor_utils import HAS_CUDA
 
 
diff --git a/test/test_fx.py b/test/test_fx.py
index 454025f98f45d0..61524358a7711e 100644
--- a/test/test_fx.py
+++ b/test/test_fx.py
@@ -57,6 +57,7 @@
     IS_WINDOWS,
     find_library_location,
     run_tests,
+    skipIfRocm,
     skipIfTorchDynamo,
 )
 from torch.testing._internal.jit_utils import JitTestCase