From 662725b936a204968abba22b212837a542d2828d Mon Sep 17 00:00:00 2001
From: Johnsonms <lizhaofu@gmail.com>
Date: Thu, 30 Oct 2025 18:56:51 -0700
Subject: [PATCH] [DeepSeekV32] Bug fix to ensure `page_table` and `result` in
 same type (#12300)

---
 python/sglang/srt/layers/attention/nsa/transform_index.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/python/sglang/srt/layers/attention/nsa/transform_index.py b/python/sglang/srt/layers/attention/nsa/transform_index.py
index 442dd113d..10b1068f5 100644
--- a/python/sglang/srt/layers/attention/nsa/transform_index.py
+++ b/python/sglang/srt/layers/attention/nsa/transform_index.py
@@ -103,7 +103,7 @@ def transform_index_page_table_decode_ref(
         result = torch.empty_like(topk_indices, dtype=torch.int32)
     assert result.shape == topk_indices.shape
     torch.gather(
-        page_table,
+        page_table.to(result.dtype),
         dim=1,
         index=topk_indices.clamp(min=0),
         out=result,