minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 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
co-authored by github-actions[bot] Jiayi Yuan
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
python/sglang/srt/batch_invariant_ops/batch_invariant_ops.py
+6 -6
View File
@@ -309,9 +309,9 @@ def matmul_persistent(
def _log_softmax_kernel( def _log_softmax_kernel(
input_ptr, input_ptr,
output_ptr, output_ptr,
input_row_stride, input_row_stride: tl.constexpr,
output_row_stride, output_row_stride: tl.constexpr,
n_cols, n_cols: tl.constexpr,
BLOCK_SIZE: tl.constexpr, BLOCK_SIZE: tl.constexpr,
): ):
""" """
@@ -814,9 +814,9 @@ def _rms_norm_kernel(
input_ptr, input_ptr,
weight_ptr, weight_ptr,
output_ptr, output_ptr,
input_row_stride, input_row_stride: tl.constexpr,
output_row_stride, output_row_stride: tl.constexpr,
n_cols, n_cols: tl.constexpr,
eps, eps,
BLOCK_SIZE: tl.constexpr, BLOCK_SIZE: tl.constexpr,
): ):