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( *,