Move swa memory pool to a seperate file (#16347)

2026-01-04 22:39:30 +08:00
parent b328cd20bb
commit 76bc07a335
14 changed files with 473 additions and 464 deletions
@@ -2,10 +2,10 @@ import unittest

 import torch

-from sglang.srt.mem_cache.allocator import SWAKVPool, SWATokenToKVPoolAllocator
 from sglang.srt.mem_cache.cache_init_params import CacheInitParams
 from sglang.srt.mem_cache.memory_pool import ReqToTokenPool
 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