[Bugfix] Fix slice operation when chunk size mismatch (#6697)

Signed-off-by: Shangming Cai <caishangming@linux.alibaba.com>
This commit is contained in:
shangmingc
2025-05-28 21:15:00 +08:00
committed by GitHub
parent 6b231325b9
commit c7588d593e

View File

@@ -354,7 +354,7 @@ class MooncakeKVManager(BaseKVManager):
kv_chunk.prefill_kv_indices kv_chunk.prefill_kv_indices
): ):
kv_chunk.prefill_kv_indices = kv_chunk.prefill_kv_indices[ kv_chunk.prefill_kv_indices = kv_chunk.prefill_kv_indices[
len(chunked_dst_kv_indice) : len(chunked_dst_kv_indice)
] ]
logger.warning( logger.warning(
f"len(chunked_dst_kv_indice) = {len(chunked_dst_kv_indice)}, len(kv_chunk.prefill_kv_indices) = {len(kv_chunk.prefill_kv_indices)}" f"len(chunked_dst_kv_indice) = {len(chunked_dst_kv_indice)}, len(kv_chunk.prefill_kv_indices) = {len(kv_chunk.prefill_kv_indices)}"