From 9c939c536aa9f1d6ff57e617e71993f10ae7cfd8 Mon Sep 17 00:00:00 2001 From: Akash Verma Date: Wed, 1 Oct 2025 21:48:41 +0000 Subject: [PATCH] Added skip for test_sac_ilp_case1 test for MI300 arch. --- test/distributed/_tools/test_sac_ilp.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/test/distributed/_tools/test_sac_ilp.py b/test/distributed/_tools/test_sac_ilp.py index 62c6afe76829..025a0f0337a1 100644 --- a/test/distributed/_tools/test_sac_ilp.py +++ b/test/distributed/_tools/test_sac_ilp.py @@ -25,6 +25,7 @@ TestCase, skipIfRocmArch, NAVI_ARCH, + MI300_ARCH, ) from torch.testing._internal.distributed._tensor.common_dtensor import ( @@ -138,7 +139,7 @@ def _collect_module_info_with_fake_tensor_mode(self) -> ModuleInfo: @skipIfTorchDynamo("https://github.com/pytorch/pytorch/issues/115653") @unittest.skipIf(not TEST_CUDA, "CUDA not available") - @skipIfRocmArch(NAVI_ARCH) + @skipIfRocmArch(NAVI_ARCH + MI300_ARCH) def test_sac_ilp_case1(self): """ This is a case where the memory budget is either binding or too tight,