[BUGFIX][Mamba] Use uint64 for address in KVBlockZeroer (#37197)

Signed-off-by: Kunshang Ji <kunshang.ji@intel.com>
2026-03-17 04:39:56 +08:00
parent 93f3c8e531
commit d157216093
1 changed files with 1 additions and 1 deletions
--- a/vllm/v1/worker/utils.py
+++ b/vllm/v1/worker/utils.py
@@ -180,7 +180,7 @@ class KVBlockZeroer:
        )
        self._ids_gpu = torch.empty(self._id_cap, dtype=torch.int64, device=self.device)
        self._meta = (
-            torch.tensor(seg_addrs, dtype=torch.int64, device=self.device),
+            torch.tensor(seg_addrs, dtype=torch.uint64, device=self.device),
            page_size_el,
            blk_size,
            len(seg_addrs),