Skip to content

Commit 78f6ff7

Browse files
authored
[ROCM][release/2.6][SWDEV-526906]Added skip for test_sac_ilp_case1 test for MI300 Arch (#2689)
In this PR, I have added a skip for test_sac_ilp_case1 test for MI300 Arch in release/2.6 branch. Currently it only skips for Navi arch. There is a skip for this test for MI300 arch in release/2.8 branch. There are upstream issues for this test too https://github.com/search?q=repo%3Apytorch%2Fpytorch+test_sac_ilp_case1&type=issues. This PR is for failing test reported in Jira https://ontrack-internal.amd.com/browse/SWDEV-526906.
1 parent 0568140 commit 78f6ff7

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

test/distributed/_tools/test_sac_ilp.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@
2525
TestCase,
2626
skipIfRocmArch,
2727
NAVI_ARCH,
28+
MI300_ARCH,
2829
)
2930

3031
from torch.testing._internal.distributed._tensor.common_dtensor import (
@@ -138,7 +139,7 @@ def _collect_module_info_with_fake_tensor_mode(self) -> ModuleInfo:
138139

139140
@skipIfTorchDynamo("https://github.com/pytorch/pytorch/issues/115653")
140141
@unittest.skipIf(not TEST_CUDA, "CUDA not available")
141-
@skipIfRocmArch(NAVI_ARCH)
142+
@skipIfRocmArch(NAVI_ARCH + MI300_ARCH)
142143
def test_sac_ilp_case1(self):
143144
"""
144145
This is a case where the memory budget is either binding or too tight,

0 commit comments

Comments
 (0)