[mem_cache][3/N] refactor: move HiSparse allocators to allocator/hisparse.py (#26678)
Co-authored-by: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
This commit is contained in:
co-authored by
Claude Opus 4.8
parent
be45745f38
commit
66076f2409
@@ -94,10 +94,10 @@ class TestHiSparseUnit(unittest.TestCase):
|
||||
|
||||
global_page_size = 1 if is_hip() else PAGE_SIZE
|
||||
|
||||
from sglang.srt.mem_cache.hisparse_memory_pool import (
|
||||
HiSparseDSATokenToKVPool,
|
||||
from sglang.srt.mem_cache.allocator.hisparse import (
|
||||
HiSparseTokenToKVPoolAllocator,
|
||||
)
|
||||
from sglang.srt.mem_cache.hisparse_memory_pool import HiSparseDSATokenToKVPool
|
||||
|
||||
cls.device_pool = HiSparseDSATokenToKVPool(
|
||||
size=SIZE,
|
||||
|
||||
Reference in New Issue
Block a user