[CI] Graceful teardown for the PD and HiSparse server fixtures (#37485)

This commit is contained in:
Alison Shao
2026-09-02 17:22:18 -07:00
committed by GitHub
parent ff04a00d73
commit db1eb48651
6 changed files with 21 additions and 24 deletions
@@ -26,7 +26,6 @@ import unittest
from types import SimpleNamespace
from urllib.parse import urlparse
from sglang.srt.utils import kill_process_tree
from sglang.test.ci.ci_register import register_cuda_ci
from sglang.test.kits.kl_divergence_kit import KLDivergenceMixin
from sglang.test.server_fixtures.default_fixture import (
@@ -36,6 +35,7 @@ from sglang.test.server_fixtures.default_fixture import (
from sglang.test.test_utils import (
DEFAULT_HYBRID_MAMBA_MODEL_NAME_FOR_TEST,
popen_launch_server,
terminate_and_kill_process_tree,
)
register_cuda_ci(est_time=800, stage="extra-b", runner_config="4-gpu-h100")
@@ -115,7 +115,7 @@ class TestUnifiedRadixTreeInt8MambaCheckpointE2E(TestInt8MambaCheckpointE2E):
@classmethod
def tearDownClass(cls):
kill_process_tree(cls.process.pid, wait_timeout=60)
terminate_and_kill_process_tree(cls.process, wait_timeout=60)
time.sleep(2)
@@ -7,7 +7,6 @@ hits return another rank's KV. The KL cases catch that as a large divergence.
Blackwell-only: the MLA DCP decode path needs ``tokenspeed_mla`` (SM100/12x).
"""
import subprocess
import unittest
from sglang.test.ci.ci_register import register_cuda_ci
@@ -99,11 +98,6 @@ class TestUnifiedKimiLinearDcpHiCache(UnifiedRadixTreeTestMixin, CustomTestCase)
@classmethod
def tearDownClass(cls):
cls.process.terminate()
try:
cls.process.wait(timeout=60)
except subprocess.TimeoutExpired:
pass
terminate_and_kill_process_tree(cls.process, wait_timeout=60)