From 36ba60ad49bcd00a0b38924c30b35551c5c27923 Mon Sep 17 00:00:00 2001 From: Ke Bao Date: Mon, 11 May 2026 20:21:05 +0800 Subject: [PATCH] Register SWA unit tests under unit/mem_cache (#24974) --- .../unit/mem_cache}/test_swa_alloc_extend_page_estimation.py | 3 +++ .../unit/mem_cache}/test_swa_lock_release_lifecycle.py | 3 +++ 2 files changed, 6 insertions(+) rename test/{manual/dsv4 => registered/unit/mem_cache}/test_swa_alloc_extend_page_estimation.py (97%) rename test/{manual/dsv4 => registered/unit/mem_cache}/test_swa_lock_release_lifecycle.py (99%) diff --git a/test/manual/dsv4/test_swa_alloc_extend_page_estimation.py b/test/registered/unit/mem_cache/test_swa_alloc_extend_page_estimation.py similarity index 97% rename from test/manual/dsv4/test_swa_alloc_extend_page_estimation.py rename to test/registered/unit/mem_cache/test_swa_alloc_extend_page_estimation.py index 5fc6a9829..c1af18e4f 100644 --- a/test/manual/dsv4/test_swa_alloc_extend_page_estimation.py +++ b/test/registered/unit/mem_cache/test_swa_alloc_extend_page_estimation.py @@ -12,8 +12,11 @@ from unittest.mock import MagicMock import torch from sglang.srt.mem_cache.swa_memory_pool import SWATokenToKVPoolAllocator +from sglang.test.ci.ci_register import register_cpu_ci from sglang.test.test_utils import CustomTestCase +register_cpu_ci(est_time=2, suite="stage-a-test-cpu") + def _make_self(*, page_size: int, full_available: int, swa_available: int): full_indices = torch.tensor([10, 11], dtype=torch.int64) diff --git a/test/manual/dsv4/test_swa_lock_release_lifecycle.py b/test/registered/unit/mem_cache/test_swa_lock_release_lifecycle.py similarity index 99% rename from test/manual/dsv4/test_swa_lock_release_lifecycle.py rename to test/registered/unit/mem_cache/test_swa_lock_release_lifecycle.py index 4be0247e3..aa6f82629 100644 --- a/test/manual/dsv4/test_swa_lock_release_lifecycle.py +++ b/test/registered/unit/mem_cache/test_swa_lock_release_lifecycle.py @@ -27,8 +27,11 @@ from sglang.srt.mem_cache.radix_cache import RadixKey from sglang.srt.mem_cache.swa_memory_pool import SWAKVPool, SWATokenToKVPoolAllocator from sglang.srt.mem_cache.swa_radix_cache import SWARadixCache from sglang.srt.utils import get_device +from sglang.test.ci.ci_register import register_cuda_ci from sglang.test.test_utils import CustomTestCase +register_cuda_ci(est_time=12, suite="stage-b-test-1-gpu-small") + def _build_tree( *,