From 1e80d938ff9a460ed3bc7f3c8e08e3d9e3402d85 Mon Sep 17 00:00:00 2001
From: huangtingwei <141888744+huangtingwei9988@users.noreply.github.com>
Date: Wed, 1 Jul 2026 19:31:16 +0800
Subject: [PATCH] [HiCache]fix draft host pool allocator type (#29823)

---
 python/sglang/srt/mem_cache/kv_cache_builder.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/python/sglang/srt/mem_cache/kv_cache_builder.py b/python/sglang/srt/mem_cache/kv_cache_builder.py
index 5788a67bc..6b376f323 100644
--- a/python/sglang/srt/mem_cache/kv_cache_builder.py
+++ b/python/sglang/srt/mem_cache/kv_cache_builder.py
@@ -108,6 +108,7 @@ def maybe_register_hicache_draft(
         host_size=0,
         page_size=page_size,
         layout=server_args.hicache_mem_layout,
+        allocator_type=server_args.hicache_storage_backend,
     )
     if isinstance(pool, MHATokenToKVPool):
         draft_host_pool = get_mha_host_pool_cls(pool)(pool, **kw)