Skipped sdpa_2 test in test_aot_inductor

Author: iupaikov-amd

test/inductor/test_aot_inductor.py

@@ -44,8 +44,10 @@
     IS_MACOS,
     IS_WINDOWS,
     skipIfRocm,
+    skipIfRocmArch,
     skipIfXpu,
     TEST_WITH_ROCM,
+    NAVI32_ARCH,
 )
 from torch.testing._internal.inductor_utils import GPU_TYPE
 from torch.testing._internal.logging_utils import LoggingTestCase, make_logging_test
@@ -925,6 +927,8 @@ def forward(self, q, k, v):
         )
         self.check_model(Model(), example_inputs)
 
+    # Eager mode produces incorrect tensor values for navi32 during this test
+    @skipIfRocmArch(NAVI32_ARCH)
     @unittest.skipIf(IS_FBCODE, "Not yet runnable in fbcode")
     @unittest.skipIf(not SM80OrLater, "bfloat16 only supported in sm80+")
     def test_sdpa_2(self):

torch/testing/_internal/common_utils.py

@@ -1360,6 +1360,8 @@ def printErrors(self) -> None:
 IS_ARM64 = platform.machine() in ('arm64', 'aarch64')
 IS_S390X = platform.machine() == "s390x"
 
+NAVI32_ARCH = "gfx1101"
+
 def is_navi_arch():
     if torch.cuda.is_available():
         prop = torch.cuda.get_device_properties(0)