ci: unify PR test suite naming (#21187)
This commit is contained in:
@@ -13,8 +13,8 @@ from sglang.srt.utils import get_device, get_device_capability, is_hip
|
||||
from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
|
||||
from sglang.test.test_utils import CustomTestCase, empty_gpu_cache
|
||||
|
||||
register_cuda_ci(est_time=80, suite="stage-b-test-large-1-gpu")
|
||||
register_amd_ci(est_time=30, suite="stage-b-test-small-1-gpu-amd")
|
||||
register_cuda_ci(est_time=80, suite="stage-b-test-1-gpu-large")
|
||||
register_amd_ci(est_time=30, suite="stage-b-test-1-gpu-small-amd")
|
||||
|
||||
_is_hip = is_hip()
|
||||
_is_fp8_fnuz = is_fp8_fnuz()
|
||||
|
||||
@@ -11,7 +11,7 @@ from sglang.test.test_utils import (
|
||||
popen_launch_server,
|
||||
)
|
||||
|
||||
register_cuda_ci(est_time=100, suite="stage-b-test-large-2-gpu")
|
||||
register_cuda_ci(est_time=100, suite="stage-b-test-2-gpu-large")
|
||||
|
||||
|
||||
class TestGLM4MoE(CustomTestCase):
|
||||
|
||||
@@ -12,7 +12,7 @@ from sglang.test.test_utils import (
|
||||
popen_launch_server,
|
||||
)
|
||||
|
||||
register_cuda_ci(est_time=140, suite="stage-b-test-large-2-gpu")
|
||||
register_cuda_ci(est_time=140, suite="stage-b-test-2-gpu-large")
|
||||
|
||||
|
||||
class TestEp(CustomTestCase):
|
||||
|
||||
@@ -15,8 +15,8 @@ from sglang.test.test_utils import (
|
||||
popen_launch_server,
|
||||
)
|
||||
|
||||
register_cuda_ci(est_time=210, suite="stage-b-test-large-1-gpu")
|
||||
register_amd_ci(est_time=1400, suite="stage-b-test-small-1-gpu-amd")
|
||||
register_cuda_ci(est_time=210, suite="stage-b-test-1-gpu-large")
|
||||
register_amd_ci(est_time=1400, suite="stage-b-test-1-gpu-small-amd")
|
||||
|
||||
|
||||
class TestTorchCompileMoe(CustomTestCase):
|
||||
|
||||
@@ -12,7 +12,7 @@ from sglang.srt.server_args import ServerArgs, set_global_server_args_for_schedu
|
||||
from sglang.test.ci.ci_register import register_cuda_ci
|
||||
from sglang.test.test_utils import CustomTestCase
|
||||
|
||||
register_cuda_ci(est_time=89, suite="stage-b-test-large-1-gpu")
|
||||
register_cuda_ci(est_time=89, suite="stage-b-test-1-gpu-large")
|
||||
|
||||
|
||||
class TestFusedMOE(CustomTestCase):
|
||||
|
||||
@@ -11,7 +11,7 @@ from sglang.srt.server_args import ServerArgs, set_global_server_args_for_schedu
|
||||
from sglang.test.ci.ci_register import register_cuda_ci
|
||||
from sglang.test.test_utils import CustomTestCase
|
||||
|
||||
register_cuda_ci(est_time=16, suite="stage-b-test-large-1-gpu")
|
||||
register_cuda_ci(est_time=16, suite="stage-b-test-1-gpu-large")
|
||||
|
||||
|
||||
def native_w8a8_per_token_matmul(A, B, As, Bs, output_dtype=torch.float16):
|
||||
|
||||
Reference in New Issue
Block a user