[claudesquad] update from 'fix-ga-perf-2' on 09 Jan 26 14:08 CST

This commit is contained in:
Zijie Tian
2026-01-09 14:08:12 +08:00
parent 79c4df4a27
commit 47e3e465f0
4 changed files with 628 additions and 278 deletions

View File

@@ -36,10 +36,11 @@ def create_kvcache_manager(config: "Config") -> KVCacheManager:
KVCacheManager instance
"""
if not getattr(config, 'enable_cpu_offload', False):
# Default: pure GPU mode
# Default: pure GPU mode with contiguous cache for single-seq optimization
return GPUOnlyManager(
num_blocks=config.num_kvcache_blocks,
block_size=config.kvcache_block_size,
max_seq_len=config.max_model_len, # Enable contiguous cache
)
# CPU offload is enabled