[AMD] fix amd ci dpskv32 (#17432)

Co-authored-by: michaelzhang-ai <michaelzhang.ai@users.noreply.github.com>
This commit is contained in:
YC Tseng
2026-01-21 20:34:24 -08:00
committed by GitHub
co-authored by michaelzhang-ai
parent fafa171529
commit 17807caf82
5 changed files with 16 additions and 7 deletions
@@ -15,10 +15,16 @@ from sglang.test.test_utils import (
write_github_step_summary,
)
register_amd_ci(est_time=3600, suite="stage-c-test-large-8-gpu-amd-mi35x")
register_amd_ci(
est_time=3600,
suite="stage-c-test-large-8-gpu-amd-mi35x",
disabled="move to nightly for saving time",
)
DEEPSEEK_V32_MODEL_PATH = "deepseek-ai/DeepSeek-V3.2"
@unittest.skipIf(is_in_amd_ci(), "Skip DP test for AMD CI, run TP only.")
class TestDeepseekV32DP(CustomTestCase):
@classmethod
def setUpClass(cls):
@@ -90,7 +96,6 @@ class TestDeepseekV32DP(CustomTestCase):
self.assertGreater(speed, 50)
@unittest.skipIf(is_in_amd_ci(), "To reduce the CI execution time for AMD.")
class TestDeepseekV32TP(CustomTestCase):
@classmethod
def setUpClass(cls):