---
id: pr-sglang-18762
repo: sgl-project/sglang
pr: 18762
title: '[diffusion] Diffusion norm fusion for z-image'
author: qimcis
date: '2026-04-04'
url: https://github.com/sgl-project/sglang/pull/18762
source_category: upstream-code
architectures:
- sm100
tags:
- benchmark_test
- compiler_runtime
- fused norm
- norm_elementwise
techniques:
- kernel-fusion
hardware_features: []
kernel_types:
- compiler
- norm
languages:
- cuda-cpp
- cute-dsl
captured_at: '2026-05-16'
status: merged
merge_sha: 005e582d06ed2a4d1975acf16cb5e5b20c8430cb
inclusion_reason: Look for fusion, vectorization, dtype conversion, and store-traffic reductions before changing math.
changed_paths:
- python/sglang/jit_kernel/diffusion/cutedsl/norm_tanh_mul_add_norm_scale.py
- python/sglang/jit_kernel/tests/test_norm_tanh_mul_add_norm_scale.py
- python/sglang/multimodal_gen/runtime/layers/layernorm.py
- python/sglang/multimodal_gen/runtime/models/dits/zimage.py
artifact_dir: evidence/pull-bundles/sglang/gh-18762
head_ref_oid: 9f35d61b8c9716b520aee72093f6b6b2a5ff991b
---
# [diffusion] Diffusion norm fusion for z-image

- Repository: `sgl-project/sglang`
- PR: [18762](https://github.com/sgl-project/sglang/pull/18762)
- 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/diffusion/cutedsl/norm_tanh_mul_add_norm_scale.py`
- `python/sglang/jit_kernel/tests/test_norm_tanh_mul_add_norm_scale.py`
- `python/sglang/multimodal_gen/runtime/layers/layernorm.py`
- `python/sglang/multimodal_gen/runtime/models/dits/zimage.py`

## Evidence Bundle

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