From 463baafe10454a198c14a73d6641d46bbece7c30 Mon Sep 17 00:00:00 2001 From: Lianmin Zheng Date: Fri, 20 Feb 2026 18:27:31 -0800 Subject: [PATCH] [Auto Sync] Update batch_invariant_ops.py (20260221) (#19098) Co-authored-by: github-actions[bot] Co-authored-by: Jiayi Yuan <34369239+jy-yuan@users.noreply.github.com> --- .../srt/batch_invariant_ops/batch_invariant_ops.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/python/sglang/srt/batch_invariant_ops/batch_invariant_ops.py b/python/sglang/srt/batch_invariant_ops/batch_invariant_ops.py index b71fc97db..80bb66e7e 100644 --- a/python/sglang/srt/batch_invariant_ops/batch_invariant_ops.py +++ b/python/sglang/srt/batch_invariant_ops/batch_invariant_ops.py @@ -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, ):