Skip to content

Commit a5f845a

Browse files
committed
change dtype of indices
1 parent 7cdd181 commit a5f845a

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

vllm/v1/worker/hpu_model_runner.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2485,9 +2485,9 @@ def _make_src_and_dst_indices(
24852485

24862486
for idx in range(len(src_block_ids)):
24872487
src_block_id = src_block_ids[idx]
2488-
src_indices = torch.range(block_size * src_block_id, block_size * (1 + src_block_id))
2488+
src_indices = torch.range(block_size * src_block_id, block_size * (1 + src_block_id), dtype=torch.int)
24892489
dst_block_id = dst_block_ids[idx]
2490-
dst_indices = torch.range(block_size * dst_block_id, block_size * (1 + dst_block_id))
2490+
dst_indices = torch.range(block_size * dst_block_id, block_size * (1 + dst_block_id), dtype=torch.int)
24912491

24922492

24932493
return src_indices, dst_indices

0 commit comments

Comments
 (0)