chore: update CI test est_time values (#22565)
Co-authored-by: Cheng Wan <chwan@rice.edu> Co-authored-by: sglang-bot <sglang-bot@users.noreply.github.com> Co-authored-by: Cheng Wan <54331508+ch-wan@users.noreply.github.com>
This commit is contained in:
co-authored by
Cheng Wan
sglang-bot
Cheng Wan
parent
b5e4ae7b1a
commit
265696b176
@@ -15,7 +15,7 @@ from sglang.srt.server_args 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=9, suite="stage-b-test-1-gpu-small")
|
||||
register_cuda_ci(est_time=8, suite="stage-b-test-1-gpu-small")
|
||||
register_amd_ci(est_time=15, suite="stage-b-test-1-gpu-small-amd")
|
||||
|
||||
|
||||
|
||||
@@ -10,7 +10,7 @@ import sglang as sgl
|
||||
from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
|
||||
from sglang.test.test_utils import CustomTestCase
|
||||
|
||||
register_cuda_ci(est_time=90, suite="stage-b-test-1-gpu-large")
|
||||
register_cuda_ci(est_time=78, suite="stage-b-test-1-gpu-large")
|
||||
register_amd_ci(est_time=90, suite="stage-b-test-1-gpu-small-amd")
|
||||
|
||||
MODEL_PATH = "Qwen/Qwen3-0.6B"
|
||||
|
||||
@@ -20,7 +20,7 @@ from sglang.test.test_utils import (
|
||||
find_available_port,
|
||||
)
|
||||
|
||||
register_cuda_ci(est_time=64, suite="stage-c-test-4-gpu-h100")
|
||||
register_cuda_ci(est_time=57, suite="stage-c-test-4-gpu-h100")
|
||||
register_amd_ci(
|
||||
est_time=64,
|
||||
suite="stage-c-test-4-gpu-amd",
|
||||
|
||||
@@ -12,7 +12,7 @@ from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
|
||||
register_amd_ci(
|
||||
est_time=19, suite="stage-b-test-2-gpu-large-amd", disabled="see #11127"
|
||||
)
|
||||
register_cuda_ci(est_time=19, suite="stage-b-test-2-gpu-large")
|
||||
register_cuda_ci(est_time=22, suite="stage-b-test-2-gpu-large")
|
||||
|
||||
|
||||
class TestReleaseMemoryOccupation(unittest.TestCase):
|
||||
|
||||
@@ -9,7 +9,7 @@ import torch
|
||||
from sglang.test.ci.ci_register import register_cpu_ci
|
||||
from sglang.test.test_utils import CustomTestCase
|
||||
|
||||
register_cpu_ci(est_time=5, suite="stage-a-test-cpu")
|
||||
register_cpu_ci(est_time=6, suite="stage-a-test-cpu")
|
||||
|
||||
|
||||
# ---------------------------------------------------------------------------
|
||||
|
||||
@@ -22,7 +22,7 @@ from sglang.test.test_utils import (
|
||||
popen_launch_server,
|
||||
)
|
||||
|
||||
register_cuda_ci(est_time=200, suite="stage-b-test-2-gpu-large")
|
||||
register_cuda_ci(est_time=164, suite="stage-b-test-2-gpu-large")
|
||||
register_amd_ci(
|
||||
est_time=200,
|
||||
suite="stage-b-test-2-gpu-large-amd",
|
||||
|
||||
@@ -1,6 +1,6 @@
|
||||
from sglang.test.ci.ci_register import register_cuda_ci
|
||||
|
||||
register_cuda_ci(est_time=100, suite="stage-c-test-4-gpu-b200")
|
||||
register_cuda_ci(est_time=149, suite="stage-c-test-4-gpu-b200")
|
||||
|
||||
import unittest
|
||||
|
||||
|
||||
@@ -43,7 +43,7 @@ from sglang.test.test_utils import (
|
||||
)
|
||||
from sglang.utils import terminate_process
|
||||
|
||||
register_cuda_ci(est_time=103, suite="stage-b-test-2-gpu-large")
|
||||
register_cuda_ci(est_time=91, suite="stage-b-test-2-gpu-large")
|
||||
register_amd_ci(est_time=103, suite="stage-b-test-2-gpu-large-amd")
|
||||
|
||||
mp.set_start_method("spawn", force=True)
|
||||
|
||||
@@ -1,6 +1,6 @@
|
||||
from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
|
||||
|
||||
register_cuda_ci(est_time=195, suite="stage-b-test-1-gpu-small")
|
||||
register_cuda_ci(est_time=136, suite="stage-b-test-1-gpu-small")
|
||||
register_amd_ci(est_time=195, suite="stage-b-test-1-gpu-small-amd")
|
||||
|
||||
import gc
|
||||
|
||||
Reference in New Issue
Block a user