---
id: pr-sglang-15835
repo: sgl-project/sglang
pr: 15835
title: '[Feature] JIT Fused QK norm + qk norm clean up'
author: DarkSharpness
date: '2025-12-28'
url: https://github.com/sgl-project/sglang/pull/15835
source_category: upstream-code
architectures:
- sm100
tags:
- PDL
- benchmark_test
- compiler_runtime
- norm_elementwise
techniques:
- kernel-fusion
- warp-specialization
hardware_features:
- pdl
kernel_types:
- compiler
- moe
- norm
languages:
- cuda-cpp
captured_at: '2026-05-16'
status: merged
merge_sha: 8e43980ebbc324cf2c32a8d248fe0f54f95c339d
inclusion_reason: Mine shape sets, tolerance rules, warmup logic, and profile commands before mutating code.
changed_paths:
- python/sglang/jit_kernel/benchmark/bench_qknorm.py
- python/sglang/jit_kernel/csrc/norm.cuh
- python/sglang/jit_kernel/include/sgl_kernel/runtime.cuh
- python/sglang/jit_kernel/include/sgl_kernel/tensor.h
- python/sglang/jit_kernel/include/sgl_kernel/utils.cuh
- python/sglang/jit_kernel/include/sgl_kernel/warp.cuh
- python/sglang/jit_kernel/norm.py
- python/sglang/jit_kernel/tests/test_qknorm.py
- python/sglang/jit_kernel/utils.py
- python/sglang/srt/models/bailing_moe.py
- python/sglang/srt/models/glm4_moe.py
- python/sglang/srt/models/llada2.py
- python/sglang/srt/models/qwen3.py
- python/sglang/srt/models/qwen3_moe.py
- python/sglang/srt/models/utils.py
artifact_dir: evidence/pull-bundles/sglang/gh-15835
head_ref_oid: c796f52611832c3f3f00939b704f3951d8be11b9
---
# [Feature] JIT Fused QK norm + qk norm clean up

- Repository: `sgl-project/sglang`
- PR: [15835](https://github.com/sgl-project/sglang/pull/15835)
- Decision: `Mine shape sets, tolerance rules, warmup logic, and profile commands before mutating code.`
- Reason: Mine shape sets, tolerance rules, warmup logic, and profile commands before mutating code.

## Changed Paths

- `python/sglang/jit_kernel/benchmark/bench_qknorm.py`
- `python/sglang/jit_kernel/csrc/norm.cuh`
- `python/sglang/jit_kernel/include/sgl_kernel/runtime.cuh`
- `python/sglang/jit_kernel/include/sgl_kernel/tensor.h`
- `python/sglang/jit_kernel/include/sgl_kernel/utils.cuh`
- `python/sglang/jit_kernel/include/sgl_kernel/warp.cuh`
- `python/sglang/jit_kernel/norm.py`
- `python/sglang/jit_kernel/tests/test_qknorm.py`
- `python/sglang/jit_kernel/utils.py`
- `python/sglang/srt/models/bailing_moe.py`
- `python/sglang/srt/models/glm4_moe.py`
- `python/sglang/srt/models/llada2.py`
- `python/sglang/srt/models/qwen3.py`
- `python/sglang/srt/models/qwen3_moe.py`
- `python/sglang/srt/models/utils.py`

## Evidence Bundle

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