[HiCache] fix: apply extra_backend_tag in Mooncake batch_exists (#17265)

This commit is contained in:
fatSheep
2026-02-03 14:54:56 +08:00
committed by GitHub
parent 74f716dbd7
commit 7a9d9c79d1

View File

@@ -649,6 +649,11 @@ class MooncakeStore(HiCacheStorage):
def batch_exists(
self, keys, extra_info: Optional[HiCacheStorageExtraInfo] = None
) -> int:
# Apply extra_backend_tag prefix if available
if self.extra_backend_tag is not None:
prefix = self.extra_backend_tag
keys = [f"{prefix}_{key}" for key in keys]
if self.is_mla_backend:
query_keys = [f"{key}_{self.mla_suffix}_k" for key in keys]
key_multiplier = 1