Lightweight extract allocation logic from mem_cache/common.py to more clearly show nearly parallel variants (#29431)

This commit is contained in:
fzyzcjy
2026-07-15 14:49:37 +08:00
committed by GitHub
parent c315df49bb
commit e789ca24a7
11 changed files with 546 additions and 516 deletions
@@ -54,8 +54,8 @@ _OWNER_SITES = {
(_SB, "ScheduleBatch.prepare_for_extend", "extend_batch_idx"): 1,
(_SB, "ScheduleBatch.prepare_for_extend", "kv_committed_len"): 1,
(_SB, "ScheduleBatch.prepare_for_extend", "kv_allocated_len"): 1,
("mem_cache/common.py", "alloc_for_extend", "evict"): 1,
("mem_cache/common.py", "alloc_for_decode", "evict"): 1,
("mem_cache/allocation.py", "alloc_for_extend", "evict"): 1,
("mem_cache/allocation.py", "alloc_for_decode", "evict"): 1,
# spec v2: no pre-claim; resolve commits the full accepted run uniformly.
(*_EAGLE_DECODE, "decode_batch_idx"): 1,
(*_EAGLE_DECODE, "evict"): 1,