ci: unify PR test suite naming (#21187)
This commit is contained in:
@@ -15,8 +15,8 @@ from sglang.test.test_utils import (
|
||||
popen_launch_server,
|
||||
)
|
||||
|
||||
register_cuda_ci(est_time=73, suite="stage-b-test-large-2-gpu")
|
||||
register_amd_ci(est_time=73, suite="stage-b-test-large-2-gpu-amd")
|
||||
register_cuda_ci(est_time=73, suite="stage-b-test-2-gpu-large")
|
||||
register_amd_ci(est_time=73, suite="stage-b-test-2-gpu-large-amd")
|
||||
|
||||
|
||||
class TestDataParallelism(CustomTestCase):
|
||||
|
||||
@@ -25,7 +25,7 @@ from sglang.test.test_utils import (
|
||||
popen_launch_server,
|
||||
)
|
||||
|
||||
register_cuda_ci(est_time=350, suite="stage-b-test-large-2-gpu")
|
||||
register_cuda_ci(est_time=350, suite="stage-b-test-2-gpu-large")
|
||||
|
||||
|
||||
class TestDPAttentionDP2TP2(
|
||||
|
||||
@@ -38,8 +38,8 @@ from sglang.utils import terminate_process
|
||||
|
||||
mp.set_start_method("spawn", force=True)
|
||||
|
||||
register_cuda_ci(est_time=72, suite="stage-b-test-large-2-gpu")
|
||||
register_amd_ci(est_time=72, suite="stage-b-test-large-2-gpu-amd")
|
||||
register_cuda_ci(est_time=72, suite="stage-b-test-2-gpu-large")
|
||||
register_amd_ci(est_time=72, suite="stage-b-test-2-gpu-large-amd")
|
||||
|
||||
|
||||
def verify_params_close(params1, params2, error_msg):
|
||||
|
||||
@@ -43,7 +43,7 @@ import pytest
|
||||
|
||||
from sglang.test.ci.ci_register import register_cuda_ci
|
||||
|
||||
register_cuda_ci(est_time=5, suite="stage-b-test-small-1-gpu")
|
||||
register_cuda_ci(est_time=5, suite="stage-b-test-1-gpu-small")
|
||||
|
||||
# Import the actual parallel_state module
|
||||
parallel_state = pytest.importorskip("sglang.srt.distributed.parallel_state")
|
||||
|
||||
Reference in New Issue
Block a user