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
@@ -7,8 +7,8 @@ from typing import TYPE_CHECKING, Any, Optional

 import torch

-from sglang.srt.mem_cache.allocator import SWATokenToKVPoolAllocator
 from sglang.srt.mem_cache.base_prefix_cache import BasePrefixCache, MatchResult
+from sglang.srt.mem_cache.swa_memory_pool import SWATokenToKVPoolAllocator

 if TYPE_CHECKING:
    from sglang.srt.managers.schedule_batch import Req