gahow/agentic-kvc
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix: use sha256 (default) not sha256_cbor for block hash computation

Browse Source 
Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
This commit is contained in:
Gahow Wang
2026-05-24 01:36:05 +08:00
parent 08d5e12838
commit 0bb6a67ed3
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
third_party/vllm/vllm/distributed/kv_transfer/kv_connector/v1/mooncake/mooncake_utils.py vendored
View File

@@ -232,7 +232,7 @@ class MooncakeBootstrapServer:
after init_none_hash is called). after init_none_hash is called).
""" """
import vllm.v1.core.kv_cache_utils as kv_utils import vllm.v1.core.kv_cache_utils as kv_utils
from vllm.utils.hashing import sha256_cbor from vllm.utils.hashing import sha256
block_size = self._kv_info.get("block_size", 512) if self._kv_info else 512 block_size = self._kv_info.get("block_size", 512) if self._kv_info else 512
n = num_tokens or len(token_ids) n = num_tokens or len(token_ids)
@@ -246,7 +246,7 @@ class MooncakeBootstrapServer:
for i in range(num_blocks): for i in range(num_blocks):
block_tokens = tuple(token_ids[i * block_size:(i + 1) * block_size]) block_tokens = tuple(token_ids[i * block_size:(i + 1) * block_size])
block_hash = kv_utils.hash_block_tokens( block_hash = kv_utils.hash_block_tokens(
sha256_cbor, prev_hash, block_tokens, None) sha256, prev_hash, block_tokens, None)
prev_hash = block_hash prev_hash = block_hash
bid = self._hash_table.get(block_hash.hex()) bid = self._hash_table.get(block_hash.hex())