diff --git a/python/sglang/test/server_fixtures/disaggregation_fixture.py b/python/sglang/test/server_fixtures/disaggregation_fixture.py index 83d5d5157..6ab5ec481 100644 --- a/python/sglang/test/server_fixtures/disaggregation_fixture.py +++ b/python/sglang/test/server_fixtures/disaggregation_fixture.py @@ -23,6 +23,7 @@ logger = logging.getLogger(__name__) class PDDisaggregationServerBase(CustomTestCase): @classmethod def setUpClass(cls): + os.environ["MC_TCP_ENABLE_CONNECTION_POOL"] = "true" parsed_url = urlparse(DEFAULT_URL_FOR_TEST) cls.base_host = parsed_url.hostname base_port = str(parsed_url.port) @@ -140,6 +141,7 @@ class PDDisaggregationServerBase(CustomTestCase): @classmethod def tearDownClass(cls): + os.environ.pop("MC_TCP_ENABLE_CONNECTION_POOL") for process in [cls.process_lb, cls.process_decode, cls.process_prefill]: if process: try: diff --git a/test/registered/disaggregation/test_disaggregation_basic.py b/test/registered/disaggregation/test_disaggregation_basic.py index ea6c0bd0b..e20d59ea8 100644 --- a/test/registered/disaggregation/test_disaggregation_basic.py +++ b/test/registered/disaggregation/test_disaggregation_basic.py @@ -21,7 +21,7 @@ from sglang.test.test_utils import ( 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):