[MRV2] Avoid recompilation of _gather_block_tables_kernel (#37645)

Signed-off-by: Woosuk Kwon <woosuk@inferact.ai>
2026-03-20 10:31:45 -07:00
parent d0532bf38d
commit e80cfe575d
1 changed files with 1 additions and 1 deletions
--- a/vllm/v1/worker/gpu/block_table.py
+++ b/vllm/v1/worker/gpu/block_table.py
@@ -169,7 +169,7 @@ class BlockTables:
        return self.slot_mappings[:, :num_tokens]


-@triton.jit
+@triton.jit(do_not_specialize=["num_reqs"])
 def _gather_block_tables_kernel(
    batch_idx_to_req_idx,  # [batch_size]
    src_block_table_ptrs,  # [num_kv_cache_groups]