chenchenghao/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 11 Packages Projects Releases Wiki Activity

[Auto Sync] Update elementwise.py (20260131) (#18033)

Browse Source 
Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
This commit is contained in:
Lianmin Zheng
2026-02-01 07:39:37 -08:00
committed by GitHub
parent afebb7ab78
commit fb609669ca
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
python/sglang/srt/layers/elementwise.py
View File

@@ -230,7 +230,7 @@ def fused_rmsnorm_kernel(
hidden_dim: tl.constexpr,
BLOCK_SIZE: tl.constexpr,
):
pid = tl.program_id(axis=0)
pid = tl.program_id(axis=0).to(tl.int64)
input_start = pid * hidden_dim
offsets = tl.arange(0, BLOCK_SIZE)