From 1c5c8665594a0e58bede57301c917119968aa20b Mon Sep 17 00:00:00 2001 From: Woosuk Kwon Date: Thu, 30 Oct 2025 16:54:10 -0700 Subject: [PATCH] uint64 --- vllm/v1/worker/gpu/block_table.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/vllm/v1/worker/gpu/block_table.py b/vllm/v1/worker/gpu/block_table.py index b04cb74564..e35f0ae593 100644 --- a/vllm/v1/worker/gpu/block_table.py +++ b/vllm/v1/worker/gpu/block_table.py @@ -85,9 +85,10 @@ class BlockTables: ) def _make_ptr_tensor(self, x: Iterable[torch.Tensor]) -> torch.Tensor: + # NOTE(woosuk): Use uint64 instead of int64 to cover all possible addresses. ptrs_tensor_cpu = torch.tensor( [t.data_ptr() for t in x], - dtype=torch.int64, + dtype=torch.uint64, device="cpu", pin_memory=self.pin_memory, )