---
id: pr-sglang-20661
repo: sgl-project/sglang
pr: 20661
title: 'Fix(jit): support rmsnorm for hidden_size in {64, 128, 256}'
author: Johnsonms
date: '2026-03-23'
url: https://github.com/sgl-project/sglang/pull/20661
source_category: upstream-code
architectures:
- sm100
tags:
- compiler_runtime
- csrc kernel
- norm_elementwise
techniques: []
hardware_features: []
kernel_types:
- compiler
- norm
languages:
- cuda-cpp
captured_at: '2026-05-16'
status: merged
merge_sha: 777edb6ef7a589f4a00ebc766f4867568cae0c71
inclusion_reason: Look for fusion, vectorization, dtype conversion, and store-traffic reductions before changing math.
changed_paths:
- python/sglang/jit_kernel/csrc/elementwise/rmsnorm.cuh
- python/sglang/jit_kernel/norm.py
- python/sglang/jit_kernel/tests/test_norm_jit.py
artifact_dir: evidence/pull-bundles/sglang/gh-20661
head_ref_oid: 7034eb259d086b5d68ec8fda14452626e805a2cc
---
# Fix(jit): support rmsnorm for hidden_size in {64, 128, 256}

- Repository: `sgl-project/sglang`
- PR: [20661](https://github.com/sgl-project/sglang/pull/20661)
- Decision: `Look for fusion, vectorization, dtype conversion, and store-traffic reductions before changing math.`
- Reason: Look for fusion, vectorization, dtype conversion, and store-traffic reductions before changing math.

## Changed Paths

- `python/sglang/jit_kernel/csrc/elementwise/rmsnorm.cuh`
- `python/sglang/jit_kernel/norm.py`
- `python/sglang/jit_kernel/tests/test_norm_jit.py`

## Evidence Bundle

`evidence/pull-bundles/sglang/gh-20661`
