fix(ci): update est_time for 57 tests based on runtime analysis (#21896)

Co-authored-by: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
This commit is contained in:
Liangsheng Yin
2026-04-01 20:16:13 -07:00
committed by GitHub
co-authored by Claude Opus 4.6
parent 2ef12073f4
commit 875a615993
57 changed files with 61 additions and 59 deletions
@@ -12,7 +12,7 @@ from sglang.test.test_utils import (
DEFAULT_TARGET_MODEL_EAGLE3,
)
register_cuda_ci(est_time=50, suite="stage-b-test-1-gpu-small")
register_cuda_ci(est_time=200, suite="stage-b-test-1-gpu-small")
register_amd_ci(est_time=50, suite="stage-b-test-1-gpu-small")
_is_hip = is_hip()
@@ -12,7 +12,7 @@ from sglang.test.test_utils import (
CustomTestCase,
)
register_cuda_ci(est_time=250, suite="stage-b-test-1-gpu-large")
register_cuda_ci(est_time=450, suite="stage-b-test-1-gpu-large")
class TestEAGLEEngine(CustomTestCase):
@@ -17,7 +17,7 @@ from sglang.test.test_utils import (
)
# EAGLE with DP attention on B200 (tp=2, dp=2, requires 4 B200 GPUs)
register_cuda_ci(est_time=300, suite="stage-c-test-4-gpu-b200")
register_cuda_ci(est_time=100, suite="stage-c-test-4-gpu-b200")
def test_gsm8k(base_url: str, model: str):
@@ -9,7 +9,7 @@ from sglang.srt.speculative.eagle_utils import (
from sglang.srt.utils import get_device
from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
register_cuda_ci(est_time=3, suite="stage-b-test-1-gpu-small")
register_cuda_ci(est_time=6, suite="stage-b-test-1-gpu-small")
register_amd_ci(est_time=3, suite="stage-b-test-1-gpu-small-amd")
@@ -6,7 +6,7 @@ from sglang.srt.speculative.cpp_ngram.ngram_corpus import NgramCorpus
from sglang.test.ci.ci_register import register_cuda_ci
from sglang.test.test_utils import CustomTestCase
register_cuda_ci(est_time=30, suite="stage-b-test-1-gpu-small")
register_cuda_ci(est_time=8, suite="stage-b-test-1-gpu-small")
def _make_corpus(match_type="BFS", **kwargs):