From cef353f338437c84787c55778dbdce143bb29f69 Mon Sep 17 00:00:00 2001 From: YAMY <74099316+YAMY1234@users.noreply.github.com> Date: Sat, 21 Feb 2026 20:00:33 -0800 Subject: [PATCH] [Fix] Quick fix for int32 overflow in Mooncakes' send_kvcache_slice (#19076) --- python/sglang/srt/disaggregation/mooncake/conn.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/python/sglang/srt/disaggregation/mooncake/conn.py b/python/sglang/srt/disaggregation/mooncake/conn.py index 25c82f8a7..d0d4efd95 100644 --- a/python/sglang/srt/disaggregation/mooncake/conn.py +++ b/python/sglang/srt/disaggregation/mooncake/conn.py @@ -460,9 +460,9 @@ class MooncakeKVManager(CommonKVManager): ) return -1 - prefill_page_indices = prefill_kv_indices.reshape(-1, 1) - decode_page_indices = dst_kv_indices.reshape(-1, 1) - tokens_per_page = np.arange(page_size, dtype=np.int32).reshape(1, -1) + prefill_page_indices = prefill_kv_indices.reshape(-1, 1).astype(np.int64) + decode_page_indices = dst_kv_indices.reshape(-1, 1).astype(np.int64) + tokens_per_page = np.arange(page_size, dtype=np.int64).reshape(1, -1) bytes_per_token_on_prefill = src_kv_item_len // page_size bytes_per_token_on_decode = dst_kv_item_len // page_size src_token_slot_offsets = (