Lightweight extract allocation logic from mem_cache/common.py to more clearly show nearly parallel variants (#29431)
This commit is contained in:
@@ -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,
|
||||
|
||||
Reference in New Issue
Block a user