[AMD] Rebalance stage-c-large-8-gpu-mi35x partitions to fix 60-min timeout (#29726)
This commit is contained in:
@@ -16,7 +16,7 @@ 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=1300, suite="stage-c-test-large-8-gpu-amd-mi35x")
|
||||
|
||||
DEEPSEEK_R1_MODEL_PATH = "amd/DeepSeek-R1-MXFP4-Preview"
|
||||
SERVER_LAUNCH_TIMEOUT = 1800
|
||||
|
||||
@@ -24,7 +24,7 @@ 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=700, suite="stage-c-test-large-8-gpu-amd-mi35x")
|
||||
|
||||
KIMI_K25_MXFP4_MODEL_PATH = "amd/Kimi-K2.5-MXFP4"
|
||||
# Bumped from b071bc6f -> 419004c8 (HF main HEAD as of 2026-05-18). The pinned
|
||||
|
||||
@@ -42,7 +42,7 @@ from sglang.test.test_utils import (
|
||||
|
||||
# PR-CI AMD MI35x suite (same suite as the existing Kimi-K2.5-MXFP4 test), so
|
||||
# the PR's changed BCG code path is exercised on every run-ci.
|
||||
register_amd_ci(est_time=3600, suite="stage-c-test-large-8-gpu-amd-mi35x")
|
||||
register_amd_ci(est_time=2100, suite="stage-c-test-large-8-gpu-amd-mi35x")
|
||||
|
||||
KIMI_K25_MXFP4_MODEL_PATH = "amd/Kimi-K2.5-MXFP4"
|
||||
# Same pinned revision as test_kimi_k25_mxfp4.py: revisions from 94d8c1bd
|
||||
|
||||
@@ -25,7 +25,7 @@ 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=450, suite="stage-c-test-large-8-gpu-amd-mi35x")
|
||||
|
||||
QWEN3_CODER_NEXT_MODEL_PATH = "Qwen/Qwen3-Coder-Next"
|
||||
SERVER_LAUNCH_TIMEOUT = 1800
|
||||
|
||||
@@ -24,7 +24,7 @@ from sglang.test.test_utils import (
|
||||
|
||||
# CI Registration — large suite to fit the integration test's server startup.
|
||||
register_cuda_ci(est_time=79, stage="base-b", runner_config="1-gpu-large")
|
||||
register_amd_ci(est_time=120, suite="stage-c-test-large-8-gpu-amd-mi35x")
|
||||
register_amd_ci(est_time=200, suite="stage-c-test-large-8-gpu-amd-mi35x")
|
||||
|
||||
|
||||
class TestBreakableCUDAGraphBasic(CustomTestCase):
|
||||
|
||||
Reference in New Issue
Block a user