Skip to content

Commit 6ca2a67

Browse files
jianzsdbyoung18
authored andcommitted
[Misc] Remove redundant comment (vllm-project#16703)
Signed-off-by: Jade Zheng <[email protected]>
1 parent 7d64cac commit 6ca2a67

File tree

1 file changed

+0
-3
lines changed

1 file changed

+0
-3
lines changed

vllm/v1/worker/gpu_model_runner.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -540,9 +540,6 @@ def _prepare_inputs(
540540
# because M (max_model_len) is not necessarily divisible by block_size.
541541
block_table_indices = (req_indices * self.max_num_blocks_per_req +
542542
positions_np // self.block_size)
543-
# NOTE(woosuk): We use torch.index_select instead of np.take here
544-
# because torch.index_select is much faster than np.take for large
545-
# tensors.
546543
block_table_cpu = self.input_batch.block_table.get_cpu_tensor()
547544
block_numbers = block_table_cpu.flatten()[block_table_indices].numpy()
548545
block_offsets = positions_np % self.block_size

0 commit comments

Comments
 (0)