---
id: pr-sglang-19437
repo: sgl-project/sglang
pr: 19437
title: '[Kernel Slimming] Migrate NVFP4 kernels to JIT'
author: mmangkad
date: '2026-03-05'
url: https://github.com/sgl-project/sglang/pull/19437
source_category: upstream-code
architectures:
- sm100
tags:
- fp4
- gemm
- moe
- nvfp4
- quantization
techniques: []
hardware_features:
- fp4
- nvfp4
kernel_types:
- gemm
- moe
- quantization
languages:
- cuda-cpp
- python
captured_at: '2026-04-27'
status: merged
merge_sha: 2bdd89a6cd6e32586c62c8b14df348f8b373fb30
inclusion_reason: kernel file changes
changed_paths:
- python/sglang/jit_kernel/benchmark/bench_nvfp4_blockwise_moe.py
- python/sglang/jit_kernel/benchmark/bench_nvfp4_quant.py
- python/sglang/jit_kernel/benchmark/bench_nvfp4_scaled_mm.py
- python/sglang/jit_kernel/csrc/gemm/nvfp4/nvfp4_expert_quant.cuh
- python/sglang/jit_kernel/csrc/gemm/nvfp4/nvfp4_quant.cuh
- python/sglang/jit_kernel/csrc/gemm/nvfp4/nvfp4_quant_entry.cuh
- python/sglang/jit_kernel/csrc/gemm/nvfp4/nvfp4_quant_kernels.cuh
- python/sglang/jit_kernel/csrc/gemm/nvfp4/nvfp4_scaled_mm_entry.cuh
- python/sglang/jit_kernel/csrc/gemm/nvfp4/nvfp4_scaled_mm_kernels.cuh
- python/sglang/jit_kernel/csrc/moe/nvfp4_blockwise_moe.cuh
- python/sglang/jit_kernel/nvfp4.py
- python/sglang/jit_kernel/tests/test_nvfp4_blockwise_moe.py
- python/sglang/jit_kernel/tests/test_nvfp4_gemm.py
- python/sglang/jit_kernel/tests/test_nvfp4_quant.py
- python/sglang/srt/layers/moe/cutlass_moe.py
- python/sglang/srt/layers/moe/cutlass_moe_params.py
- python/sglang/srt/layers/moe/moe_runner/flashinfer_trtllm.py
- python/sglang/srt/layers/moe/token_dispatcher/standard.py
- python/sglang/srt/layers/quantization/modelopt_quant.py
- sgl-kernel/CMakeLists.txt
- sgl-kernel/csrc/common_extension.cc
- sgl-kernel/csrc/gemm/nvfp4_quant_entry.cu
- sgl-kernel/csrc/gemm/nvfp4_scaled_mm_entry.cu
- sgl-kernel/include/sgl_kernel_ops.h
- sgl-kernel/python/sgl_kernel/__init__.py
- sgl-kernel/python/sgl_kernel/gemm.py
- sgl-kernel/python/sgl_kernel/moe.py
head_ref_oid: a3bce4ce788bb6a11def8c94e93903846cb8269e
artifact_dir: evidence/pull-bundles/sglang/gh-19437
---
# [Kernel Slimming] Migrate NVFP4 kernels to JIT

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

## Changed Paths

- `python/sglang/jit_kernel/benchmark/bench_nvfp4_blockwise_moe.py`
- `python/sglang/jit_kernel/benchmark/bench_nvfp4_quant.py`
- `python/sglang/jit_kernel/benchmark/bench_nvfp4_scaled_mm.py`
- `python/sglang/jit_kernel/csrc/gemm/nvfp4/nvfp4_expert_quant.cuh`
- `python/sglang/jit_kernel/csrc/gemm/nvfp4/nvfp4_quant.cuh`
- `python/sglang/jit_kernel/csrc/gemm/nvfp4/nvfp4_quant_entry.cuh`
- `python/sglang/jit_kernel/csrc/gemm/nvfp4/nvfp4_quant_kernels.cuh`
- `python/sglang/jit_kernel/csrc/gemm/nvfp4/nvfp4_scaled_mm_entry.cuh`
- `python/sglang/jit_kernel/csrc/gemm/nvfp4/nvfp4_scaled_mm_kernels.cuh`
- `python/sglang/jit_kernel/csrc/moe/nvfp4_blockwise_moe.cuh`
- `python/sglang/jit_kernel/nvfp4.py`
- `python/sglang/jit_kernel/tests/test_nvfp4_blockwise_moe.py`
- `python/sglang/jit_kernel/tests/test_nvfp4_gemm.py`
- `python/sglang/jit_kernel/tests/test_nvfp4_quant.py`
- `python/sglang/srt/layers/moe/cutlass_moe.py`
- `python/sglang/srt/layers/moe/cutlass_moe_params.py`
- `python/sglang/srt/layers/moe/moe_runner/flashinfer_trtllm.py`
- `python/sglang/srt/layers/moe/token_dispatcher/standard.py`
- `python/sglang/srt/layers/quantization/modelopt_quant.py`
- `sgl-kernel/CMakeLists.txt`
- `sgl-kernel/csrc/common_extension.cc`
- `sgl-kernel/csrc/gemm/nvfp4_quant_entry.cu`
- `sgl-kernel/csrc/gemm/nvfp4_scaled_mm_entry.cu`
- `sgl-kernel/include/sgl_kernel_ops.h`
- `sgl-kernel/python/sgl_kernel/__init__.py`
- `sgl-kernel/python/sgl_kernel/gemm.py`
- `sgl-kernel/python/sgl_kernel/moe.py`

## Evidence Bundle

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