Imported skipIfRocm in certain test suites (#1577)

BLOrange-AMD · web-flow · commit e9b772791750 · 2024-09-08T22:47:51.000-07:00
Fixes SWDEV-472397
diff --git 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
@@ -57,6 +57,7 @@
     IS_WINDOWS,
     find_library_location,
     run_tests,
+    skipIfRocm,
     skipIfTorchDynamo,
 )
 from torch.testing._internal.jit_utils import JitTestCase

Original file line number	Diff line number	Diff line change
`@@ -57,6 +57,7 @@`
`57`	`57`	`IS_WINDOWS,`
`58`	`58`	`find_library_location,`
`59`	`59`	`run_tests,`
	`60`	`+ skipIfRocm,`
`60`	`61`	`skipIfTorchDynamo,`
`61`	`62`	`)`
`62`	`63`	`from torch.testing._internal.jit_utils import JitTestCase`