[HiCache] Account for newly pinned ancestors in load-back quota (#38481)

This commit is contained in:
Shuwen Wang
2026-09-10 23:54:08 +08:00
committed by GitHub
parent 92dffebe16
commit cc7e43ad22
2 changed files with 3 additions and 2 deletions
@@ -1594,7 +1594,7 @@ class UnifiedRadixCache(BasePrefixCache):
# entirely empty spec (e.g. foreign-pin rejection) must never report
# success, even at load_back_threshold <= 0.
if (kv_tokens < max(1, self.load_back_threshold) and not comp_xfers) or (
mem_quota is not None and kv_tokens > mem_quota + result.delta
mem_quota is not None and kv_tokens + result.delta > mem_quota
):
self.dec_lock_ref(node_id, ancestor_lock_params)
self.dec_host_lock_ref(node_id, host_anchor_params)