[HiCache] Fix side pools to use resolved host allocator (#38350)

2026-09-10 14:28:20 +08:00
parent 69777c4d36
commit 058cca0192
1 changed files with 2 additions and 2 deletions
@@ -918,7 +918,7 @@ def build_hybrid_mamba_swa_stack(
        mamba_pool,
        get_memory().hicache_ratio,
        mamba_host_size,
-        allocator_type=get_memory().hicache_storage_backend,
+        allocator_type=_get_allocator_type(),
        layout=get_memory().hicache_mem_layout,
    )
    entries = [
@@ -1896,7 +1896,7 @@ def build_minimax_sparse_hicache_stack(
            index_k_pool,
            kv_host_pool,
            get_memory().hicache_mem_layout,
-            allocator_type=get_memory().hicache_storage_backend,
+            allocator_type=_get_allocator_type(),
        )
        entries.append(
            build_pool_entry(