[CI bug] Fix `Each KV cache group's real block_size must be divisible by has h_block_size` (#51180)
Signed-off-by: yewentao256 <zhyanwentao@126.com>
Wentao Ye committed
Aug 5, 2026 at 14:03 UTC
23c0f337b7c25508fce2b98cbf655bdc8cf166b4
1 file changed
+1
-4
vllm/v1/core/kv_cache_utils.py
+1
-4
@@ -645,12 +645,9 @@ def resolve_kv_cache_block_sizes(
645
if not (cache_config.enable_prefix_caching or connector_enabled):
646
return scheduler_block_size, scheduler_block_size
647
648
- # Mamba groups with block_size != cache_config.block_size
649
- # (mamba_cache_mode != "align") break divisibility; back off to the
650
- # scheduler block size.
648
if any(
649
isinstance(g.kv_cache_spec, MambaSpec)
653
- and g.kv_cache_spec.block_size != cache_config.block_size
650
+ and g.kv_cache_spec.mamba_cache_mode != "align"
651
for g in groups
652
):
653
return scheduler_block_size, scheduler_block_size