Skip to content

GitLab

Unverified Commit 662725b9 authored by Johnsonms's avatar Johnsonms Committed by GitHub
Browse files

[DeepSeekV32] Bug fix to ensure `page_table` and `result` in same type (#12300)

parent 82cfcd3b
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
python/sglang/srt/layers/attention/nsa/transform_index.py
View file @ 662725b9
...@@ -103,7 +103,7 @@ def transform_index_page_table_decode_ref( ...@@ -103,7 +103,7 @@ def transform_index_page_table_decode_ref(
result = torch.empty_like(topk_indices, dtype=torch.int32) result = torch.empty_like(topk_indices, dtype=torch.int32)
assert result.shape == topk_indices.shape assert result.shape == topk_indices.shape
torch.gather( torch.gather(
page_table, page_table.to(result.dtype),
dim=1, dim=1,
index=topk_indices.clamp(min=0), index=topk_indices.clamp(min=0),
out=result, out=result,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment