[NO CP] Updated changes to skip few UTs

Author: pruthvistony

test/dynamo/test_structured_trace.py

@@ -161,6 +161,7 @@ def test_schedule(self):
         self.assertParses()
 
     @requires_cuda
+    @skipIfRocm(msg="TODO: temp skip on ROCm 6.2")
     def test_cudagraphs(self):
         fn_opt = torch.compile(mode="reduce-overhead")(inductor_schedule_fn)
         fn_opt(torch.ones(1000, 1000, device="cuda"))

test/inductor/test_torchinductor_dynamic_shapes.py

@@ -195,6 +195,7 @@ def fn(x, y):
         self.assertEqual(r, opt_r)
 
     @torch._dynamo.config.patch(capture_scalar_outputs=True)
+    @skipIfRocm(msg="TODO: temp skip on ROCm 6.2")
     def test_unwrap_storage_didnt_work_repro(self, device):
         def f():
             full = torch.full((), 11)

test/run_test.py

@@ -182,6 +182,9 @@ def __contains__(self, item):
     "distributed/_tensor/test_attention",
 ]
 
+if sys.version_info.major < 3 or (sys.version_info.major == 3 and sys.version_info.minor <= 9):
+    ROCM_BLOCKLIST.append("test_typing")
+
 XPU_BLOCKLIST = [
     "test_autograd",
 ]

test/test_fx.py

@@ -4106,6 +4106,7 @@ def test_class_member_back_compat(self):
                   f"and subsequently --accept the change."
             raise AssertionError(msg) from e
 
+    @skipIfRocm(msg="TODO: flaky - https://github.com/pytorch/pytorch/issues/104012")
     def test_public_api_surface(self):
         non_back_compat_objects = {}