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

[Auto Sync] Update batch_invariant_ops.py (20260221) (#19098)

Browse Source 
Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
Co-authored-by: Jiayi Yuan <34369239+jy-yuan@users.noreply.github.com>
This commit is contained in:
Lianmin Zheng
2026-02-20 18:27:31 -08:00
committed by GitHub
parent 2928dfb8fa
commit 463baafe10
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
python/sglang/srt/batch_invariant_ops/batch_invariant_ops.py
View File

@@ -309,9 +309,9 @@ def matmul_persistent(
def _log_softmax_kernel(
input_ptr,
output_ptr,
input_row_stride,
output_row_stride,
n_cols,
input_row_stride: tl.constexpr,
output_row_stride: tl.constexpr,
n_cols: tl.constexpr,
BLOCK_SIZE: tl.constexpr,
):
"""
@@ -814,9 +814,9 @@ def _rms_norm_kernel(
input_ptr,
weight_ptr,
output_ptr,
input_row_stride,
output_row_stride,
n_cols,
input_row_stride: tl.constexpr,
output_row_stride: tl.constexpr,
n_cols: tl.constexpr,
eps,
BLOCK_SIZE: tl.constexpr,
):