[CI] Graceful teardown for kv_canary and EAGLE spec fixtures (#32829)

This commit is contained in:
Alison Shao
2026-08-02 01:09:59 -07:00
committed by GitHub
parent 8d106c3d79
commit 43be25b2b7
18 changed files with 58 additions and 46 deletions
@@ -1,12 +1,10 @@
import os
import shutil
import subprocess
import tempfile
import unittest
from test_unified_radix_cache_kl_nightly import AccuracyTwoPassMixin
from sglang.srt.utils import kill_process_tree
from sglang.test.ci.ci_register import register_cuda_ci
from sglang.test.kits.unified_radix_cache_kit import UnifiedRadixTreeTestMixin
from sglang.test.kl_multiturn_utils import (
@@ -19,6 +17,7 @@ from sglang.test.test_utils import (
DEFAULT_URL_FOR_TEST,
CustomTestCase,
popen_launch_server,
terminate_and_kill_process_tree,
)
register_cuda_ci(est_time=800, stage="extra-b", runner_config="4-gpu-h100")
@@ -69,7 +68,7 @@ class TestUnifiedMambaRadixCache(UnifiedRadixTreeTestMixin, CustomTestCase):
@classmethod
def tearDownClass(cls):
kill_process_tree(cls.process.pid)
terminate_and_kill_process_tree(cls.process, wait_timeout=60)
# ─── Mamba + HiCache L2 ──────────────────────────────────────────────────────
@@ -128,12 +127,7 @@ class TestUnifiedMambaHiCache(UnifiedRadixTreeTestMixin, CustomTestCase):
@classmethod
def tearDownClass(cls):
cls.process.terminate()
try:
cls.process.wait(timeout=60)
except subprocess.TimeoutExpired:
pass
kill_process_tree(cls.process.pid)
terminate_and_kill_process_tree(cls.process, wait_timeout=60)
# ─── Mamba + HiCache L3 (file backend) ───────────────────────────────────────
@@ -194,12 +188,7 @@ class TestUnifiedMambaHiCacheL3(AccuracyTwoPassMixin, CustomTestCase):
@classmethod
def tearDownClass(cls):
cls.process.terminate()
try:
cls.process.wait(timeout=60)
except subprocess.TimeoutExpired:
pass
kill_process_tree(cls.process.pid)
terminate_and_kill_process_tree(cls.process, wait_timeout=60)
if os.path.isdir(cls.hicache_dir):
shutil.rmtree(cls.hicache_dir, ignore_errors=True)