minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 Packages Projects Releases Wiki Activity

[CI] Move disaggregation basic CI back to 2-gpu suite (#23447)

Browse Source
This commit is contained in:
Shangming Cai
2026-04-22 17:50:33 +08:00
committed by GitHub
parent 6a3c070ee3
commit 1c06a3d072
2 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/sglang/test/server_fixtures/disaggregation_fixture.py
+2
View File
@@ -23,6 +23,7 @@ logger = logging.getLogger(__name__)
class PDDisaggregationServerBase(CustomTestCase): class PDDisaggregationServerBase(CustomTestCase):
@classmethod @classmethod
def setUpClass(cls): def setUpClass(cls):
os.environ["MC_TCP_ENABLE_CONNECTION_POOL"] = "true"
parsed_url = urlparse(DEFAULT_URL_FOR_TEST) parsed_url = urlparse(DEFAULT_URL_FOR_TEST)
cls.base_host = parsed_url.hostname cls.base_host = parsed_url.hostname
base_port = str(parsed_url.port) base_port = str(parsed_url.port)
@@ -140,6 +141,7 @@ class PDDisaggregationServerBase(CustomTestCase):
@classmethod @classmethod
def tearDownClass(cls): def tearDownClass(cls):
os.environ.pop("MC_TCP_ENABLE_CONNECTION_POOL")
for process in [cls.process_lb, cls.process_decode, cls.process_prefill]: for process in [cls.process_lb, cls.process_decode, cls.process_prefill]:
if process: if process:
try: try:
test/registered/disaggregation/test_disaggregation_basic.py
+1 -1
View File
@@ -21,7 +21,7 @@ from sglang.test.test_utils import (
DEFAULT_TARGET_MODEL_EAGLE3, DEFAULT_TARGET_MODEL_EAGLE3,
) )
register_cuda_ci(est_time=394, suite="stage-c-test-4-gpu-h100") register_cuda_ci(est_time=394, suite="stage-b-test-2-gpu-large")
class TestDisaggregationAccuracy(PauseResumeInPlaceMixin, PDDisaggregationServerBase): class TestDisaggregationAccuracy(PauseResumeInPlaceMixin, PDDisaggregationServerBase):