ci: unify PR test suite naming (#21187)
This commit is contained in:
@@ -15,8 +15,8 @@ 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-small-1-gpu")
|
||||
register_amd_ci(est_time=15, suite="stage-b-test-small-1-gpu-amd")
|
||||
register_cuda_ci(est_time=9, suite="stage-b-test-1-gpu-small")
|
||||
register_amd_ci(est_time=15, suite="stage-b-test-1-gpu-small-amd")
|
||||
|
||||
|
||||
class LMHeadStub(nn.Module):
|
||||
|
||||
@@ -10,8 +10,8 @@ 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-large-1-gpu")
|
||||
register_amd_ci(est_time=90, suite="stage-b-test-small-1-gpu-amd")
|
||||
register_cuda_ci(est_time=90, 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"
|
||||
LORA_REPO = "charent/self_cognition_Alice"
|
||||
|
||||
@@ -10,9 +10,9 @@ from sglang.srt.utils.patch_torch import monkey_patch_torch_reductions
|
||||
from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
|
||||
|
||||
register_amd_ci(
|
||||
est_time=19, suite="stage-b-test-large-2-gpu-amd", disabled="see #11127"
|
||||
est_time=19, suite="stage-b-test-2-gpu-large-amd", disabled="see #11127"
|
||||
)
|
||||
register_cuda_ci(est_time=19, suite="stage-b-test-large-2-gpu")
|
||||
register_cuda_ci(est_time=19, suite="stage-b-test-2-gpu-large")
|
||||
|
||||
|
||||
class TestReleaseMemoryOccupation(unittest.TestCase):
|
||||
|
||||
@@ -19,9 +19,9 @@ from sglang.test.test_utils import (
|
||||
)
|
||||
|
||||
register_amd_ci(
|
||||
est_time=210, suite="stage-b-test-small-1-gpu-amd", disabled="see #14021"
|
||||
est_time=210, suite="stage-b-test-1-gpu-small-amd", disabled="see #14021"
|
||||
)
|
||||
register_cuda_ci(est_time=210, suite="stage-b-test-large-1-gpu", disabled="see #14021")
|
||||
register_cuda_ci(est_time=210, suite="stage-b-test-1-gpu-large", disabled="see #14021")
|
||||
|
||||
|
||||
###############################################################################
|
||||
|
||||
@@ -43,8 +43,8 @@ from sglang.test.test_utils import (
|
||||
)
|
||||
from sglang.utils import terminate_process
|
||||
|
||||
register_cuda_ci(est_time=103, suite="stage-b-test-large-2-gpu")
|
||||
register_amd_ci(est_time=103, suite="stage-b-test-large-2-gpu-amd")
|
||||
register_cuda_ci(est_time=103, 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,7 +1,7 @@
|
||||
from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
|
||||
|
||||
register_cuda_ci(est_time=195, suite="stage-b-test-small-1-gpu")
|
||||
register_amd_ci(est_time=195, suite="stage-b-test-small-1-gpu-amd")
|
||||
register_cuda_ci(est_time=195, suite="stage-b-test-1-gpu-small")
|
||||
register_amd_ci(est_time=195, suite="stage-b-test-1-gpu-small-amd")
|
||||
|
||||
import gc
|
||||
import json
|
||||
|
||||
Reference in New Issue
Block a user