diff --git a/python/sglang/srt/managers/scheduler.py b/python/sglang/srt/managers/scheduler.py index a0c2997fe..53e9b02e3 100644 --- a/python/sglang/srt/managers/scheduler.py +++ b/python/sglang/srt/managers/scheduler.py @@ -558,7 +558,11 @@ class Scheduler( self.tree_cache = HiRadixCache( req_to_token_pool=self.req_to_token_pool, token_to_kv_pool_allocator=self.token_to_kv_pool_allocator, - tp_cache_group=self.tp_cpu_group, + tp_cache_group=( + self.attn_tp_cpu_group + if self.server_args.enable_dp_attention + else self.tp_cpu_group + ), page_size=self.page_size, hicache_ratio=server_args.hicache_ratio, hicache_size=server_args.hicache_size,