Introduce server_fixtures in sglang.test (#14899)

This commit is contained in:
Liangsheng Yin
2025-12-11 22:30:33 +09:00
committed by GitHub
parent 27032cecd9
commit 543d62d11a
9 changed files with 36 additions and 22 deletions
+1 -1
View File
@@ -3,7 +3,7 @@ from types import SimpleNamespace
from sglang.srt.utils import kill_process_tree
from sglang.test.few_shot_gsm8k import run_eval as run_eval_few_shot_gsm8k
from sglang.test.test_disaggregation_utils import get_rdma_devices_args
from sglang.test.server_fixtures.disaggregation_fixture import get_rdma_devices_args
from sglang.test.test_utils import (
DEFAULT_MODEL_NAME_FOR_TEST_MLA,
DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,
@@ -8,7 +8,9 @@ from typing import Dict
import requests
from sglang.bench_serving import get_tokenizer
from sglang.test.test_disaggregation_utils import TestDisaggregationBase
from sglang.test.server_fixtures.disaggregation_fixture import (
PDDisaggregationServerBase,
)
from sglang.test.test_utils import (
DEFAULT_MODEL_NAME_FOR_TEST,
DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,
@@ -16,7 +18,7 @@ from sglang.test.test_utils import (
)
class DisaggregationHiCacheBase(TestDisaggregationBase):
class DisaggregationHiCacheBase(PDDisaggregationServerBase):
"""Base class for disaggregation with HiCache tests"""
@classmethod
@@ -2,7 +2,9 @@ import unittest
from types import SimpleNamespace
from sglang.test.few_shot_gsm8k import run_eval as run_eval_few_shot_gsm8k
from sglang.test.test_disaggregation_utils import TestDisaggregationBase
from sglang.test.server_fixtures.disaggregation_fixture import (
PDDisaggregationServerBase,
)
from sglang.test.test_utils import (
DEFAULT_MODEL_NAME_FOR_TEST,
DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,
@@ -10,7 +12,7 @@ from sglang.test.test_utils import (
)
class TestDisaggregationPiecewiseCudaGraph(TestDisaggregationBase):
class TestDisaggregationPiecewiseCudaGraph(PDDisaggregationServerBase):
"""Test piecewise CUDA graph support in disaggregation prefill server"""
@classmethod