---
id: pr-sglang-18073
repo: sgl-project/sglang
pr: 18073
title: '[Diffsuion & JIT_kernel] QKNorm cross heads kernel'
author: BBuf
date: '2026-02-03'
url: https://github.com/sgl-project/sglang/pull/18073
source_category: upstream-code
architectures:
- sm100
tags:
- gemm
techniques: []
hardware_features: []
kernel_types: []
languages:
- cuda-cpp
- python
captured_at: '2026-04-27'
status: merged
merge_sha: a1bbc892af27867901f91e9a1c485824ff9337a6
inclusion_reason: kernel file changes
changed_paths:
- python/sglang/jit_kernel/benchmark/bench_qknorm_across_heads.py
- python/sglang/jit_kernel/csrc/elementwise/qknorm_across_heads.cuh
- python/sglang/jit_kernel/norm.py
- python/sglang/jit_kernel/tests/test_qknorm_across_heads.py
head_ref_oid: d8c3a05f488294bba72a35de12cf8f4f14d4c157
artifact_dir: evidence/pull-bundles/sglang/gh-18073
---
# [Diffsuion & JIT_kernel] QKNorm cross heads kernel

- Repository: `sgl-project/sglang`
- PR: [18073](https://github.com/sgl-project/sglang/pull/18073)
- Decision: `kernel file changes`
- Reason: kernel file changes

## Changed Paths

- `python/sglang/jit_kernel/benchmark/bench_qknorm_across_heads.py`
- `python/sglang/jit_kernel/csrc/elementwise/qknorm_across_heads.cuh`
- `python/sglang/jit_kernel/norm.py`
- `python/sglang/jit_kernel/tests/test_qknorm_across_heads.py`

## Evidence Bundle

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