ci: unify PR test suite naming (#21187)
This commit is contained in:
@@ -32,10 +32,10 @@ from sglang.test.test_utils import (
|
||||
# Embedding model tests
|
||||
register_amd_ci(
|
||||
est_time=73,
|
||||
suite="stage-b-test-small-1-gpu-amd",
|
||||
suite="stage-b-test-1-gpu-small-amd",
|
||||
disabled="see https://github.com/sgl-project/sglang/issues/11127",
|
||||
)
|
||||
register_cuda_ci(est_time=73, suite="stage-b-test-small-1-gpu")
|
||||
register_cuda_ci(est_time=73, suite="stage-b-test-1-gpu-small")
|
||||
|
||||
MODEL_TO_CONFIG = {
|
||||
"Alibaba-NLP/gte-Qwen2-1.5B-instruct": (1, 1e-5),
|
||||
|
||||
@@ -29,7 +29,7 @@ from sglang.test.test_utils import CustomTestCase, get_similarities, is_in_ci
|
||||
# python -m unittest test_encoder_embedding_models.TestEncoderEmbeddingModels.test_prefill_logits
|
||||
|
||||
|
||||
register_cuda_ci(est_time=270, suite="stage-b-test-small-1-gpu")
|
||||
register_cuda_ci(est_time=270, suite="stage-b-test-1-gpu-small")
|
||||
|
||||
MODELS = [("BAAI/bge-small-en", 1, 1e-5), ("BAAI/bge-m3", 1, 1e-5)]
|
||||
|
||||
|
||||
@@ -16,8 +16,8 @@ from sglang.test.test_utils import (
|
||||
popen_launch_server,
|
||||
)
|
||||
|
||||
register_cuda_ci(est_time=38, suite="stage-b-test-small-1-gpu")
|
||||
register_amd_ci(est_time=38, suite="stage-b-test-small-1-gpu-amd")
|
||||
register_cuda_ci(est_time=38, suite="stage-b-test-1-gpu-small")
|
||||
register_amd_ci(est_time=38, suite="stage-b-test-1-gpu-small-amd")
|
||||
|
||||
|
||||
class TestInputEmbeds(CustomTestCase):
|
||||
|
||||
@@ -30,7 +30,7 @@ from sglang.test.test_utils import (
|
||||
popen_launch_server,
|
||||
)
|
||||
|
||||
register_cuda_ci(est_time=45, suite="stage-b-test-small-1-gpu")
|
||||
register_cuda_ci(est_time=45, suite="stage-b-test-1-gpu-small")
|
||||
|
||||
CHUNKED_PREFILL_SIZE = 256
|
||||
|
||||
|
||||
@@ -13,8 +13,8 @@ from sglang.test.test_utils import (
|
||||
popen_launch_server,
|
||||
)
|
||||
|
||||
register_cuda_ci(est_time=70, suite="stage-b-test-small-1-gpu")
|
||||
register_amd_ci(est_time=141, suite="stage-b-test-small-1-gpu-amd")
|
||||
register_cuda_ci(est_time=70, suite="stage-b-test-1-gpu-small")
|
||||
register_amd_ci(est_time=141, suite="stage-b-test-1-gpu-small-amd")
|
||||
|
||||
|
||||
class TestOpenAIEmbedding(CustomTestCase):
|
||||
|
||||
Reference in New Issue
Block a user