---
id: pr-sglang-21239
repo: sgl-project/sglang
pr: 21239
title: Refactor JIT kernel CI to use run_suite.py registration system
author: merrymercy
date: '2026-03-24'
url: https://github.com/sgl-project/sglang/pull/21239
source_category: upstream-code
architectures:
- sm100
tags:
- attention
- flash-attention
- fp4
- fp8
- gemm
- kernel-fusion
- mla
- moe
- nvfp4
- quantization
techniques:
- kernel-fusion
hardware_features:
- fp4
- fp8
- nvfp4
kernel_types:
- attention
- flash-attention
- gemm
- mla
- moe
- quantization
languages:
- cute-dsl
- python
- triton
captured_at: '2026-04-17'
status: merged
merge_sha: 260abe1fb1f116ea51a72c922defccf2788197e8
inclusion_reason: kernel file changes
changed_paths:
- .github/workflows/nightly-test-nvidia.yml
- .github/workflows/pr-test-jit-kernel.yml
- python/sglang/jit_kernel/benchmark/bench_awq_dequantize.py
- python/sglang/jit_kernel/benchmark/bench_clamp_position.py
- python/sglang/jit_kernel/benchmark/bench_concat_mla.py
- python/sglang/jit_kernel/benchmark/bench_custom_all_reduce.py
- python/sglang/jit_kernel/benchmark/bench_fused_add_rmsnorm.py
- python/sglang/jit_kernel/benchmark/bench_fused_norm_scale_shift.py
- python/sglang/jit_kernel/benchmark/bench_hadamard.py
- python/sglang/jit_kernel/benchmark/bench_hicache.py
- python/sglang/jit_kernel/benchmark/bench_norm.py
- python/sglang/jit_kernel/benchmark/bench_norm_impls.py
- 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/benchmark/bench_per_tensor_quant_fp8.py
- python/sglang/jit_kernel/benchmark/bench_per_token_group_quant_8bit.py
- python/sglang/jit_kernel/benchmark/bench_qknorm.py
- python/sglang/jit_kernel/benchmark/bench_qknorm_across_heads.py
- python/sglang/jit_kernel/benchmark/bench_qwen_image_modulation.py
- python/sglang/jit_kernel/benchmark/bench_renorm.py
- python/sglang/jit_kernel/benchmark/bench_resolve_future_token_ids.py
- python/sglang/jit_kernel/benchmark/bench_rmsnorm.py
- python/sglang/jit_kernel/benchmark/bench_rope.py
- python/sglang/jit_kernel/benchmark/bench_store_cache.py
- python/sglang/jit_kernel/norm.py
- python/sglang/jit_kernel/tests/test_add_constant.py
- python/sglang/jit_kernel/tests/test_awq_dequantize.py
- python/sglang/jit_kernel/tests/test_awq_marlin_moe_repack.py
- python/sglang/jit_kernel/tests/test_awq_marlin_repack.py
- python/sglang/jit_kernel/tests/test_clamp_position.py
- python/sglang/jit_kernel/tests/test_concat_mla.py
- python/sglang/jit_kernel/tests/test_custom_all_reduce.py
- python/sglang/jit_kernel/tests/test_cutedsl_gdn.py
- python/sglang/jit_kernel/tests/test_flash_attention_4.py
- python/sglang/jit_kernel/tests/test_fused_add_rmsnorm.py
- python/sglang/jit_kernel/tests/test_fused_metadata_copy.py
- python/sglang/jit_kernel/tests/test_fused_norm_scale_shift.py
- python/sglang/jit_kernel/tests/test_fused_store_index_cache.py
- python/sglang/jit_kernel/tests/test_fused_verify_triton_gdn.py
- python/sglang/jit_kernel/tests/test_gptq_marlin.py
- python/sglang/jit_kernel/tests/test_gptq_marlin_repack.py
- python/sglang/jit_kernel/tests/test_hadamard_jit.py
- python/sglang/jit_kernel/tests/test_moe_lora_align_block_size.py
- python/sglang/jit_kernel/tests/test_moe_wna16_marlin.py
- python/sglang/jit_kernel/tests/test_norm_jit.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/jit_kernel/tests/test_per_tensor_quant_fp8.py
- python/sglang/jit_kernel/tests/test_per_token_group_quant_8bit.py
- python/sglang/jit_kernel/tests/test_pos_enc.py
- python/sglang/jit_kernel/tests/test_qknorm.py
- python/sglang/jit_kernel/tests/test_qknorm_across_heads.py
- python/sglang/jit_kernel/tests/test_qwen_image_modulation.py
- python/sglang/jit_kernel/tests/test_renorm.py
- python/sglang/jit_kernel/tests/test_resolve_future_token_ids.py
- python/sglang/jit_kernel/tests/test_rmsnorm.py
- python/sglang/jit_kernel/tests/test_rope.py
- python/sglang/jit_kernel/tests/test_store_cache.py
- python/sglang/jit_kernel/tests/test_timestep_embedding.py
- scripts/version_branch_to_tag.sh
- test/run_suite.py
artifact_dir: evidence/pull-bundles/sglang/gh-21239
head_ref_oid: 215f6fae42e866feec6ad997c9bc620581a6dd4a
---
# Refactor JIT kernel CI to use run_suite.py registration system

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

## Changed Paths

- `.github/workflows/nightly-test-nvidia.yml`
- `.github/workflows/pr-test-jit-kernel.yml`
- `python/sglang/jit_kernel/benchmark/bench_awq_dequantize.py`
- `python/sglang/jit_kernel/benchmark/bench_clamp_position.py`
- `python/sglang/jit_kernel/benchmark/bench_concat_mla.py`
- `python/sglang/jit_kernel/benchmark/bench_custom_all_reduce.py`
- `python/sglang/jit_kernel/benchmark/bench_fused_add_rmsnorm.py`
- `python/sglang/jit_kernel/benchmark/bench_fused_norm_scale_shift.py`
- `python/sglang/jit_kernel/benchmark/bench_hadamard.py`
- `python/sglang/jit_kernel/benchmark/bench_hicache.py`
- `python/sglang/jit_kernel/benchmark/bench_norm.py`
- `python/sglang/jit_kernel/benchmark/bench_norm_impls.py`
- `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/benchmark/bench_per_tensor_quant_fp8.py`
- `python/sglang/jit_kernel/benchmark/bench_per_token_group_quant_8bit.py`
- `python/sglang/jit_kernel/benchmark/bench_qknorm.py`
- `python/sglang/jit_kernel/benchmark/bench_qknorm_across_heads.py`
- `python/sglang/jit_kernel/benchmark/bench_qwen_image_modulation.py`
- `python/sglang/jit_kernel/benchmark/bench_renorm.py`
- `python/sglang/jit_kernel/benchmark/bench_resolve_future_token_ids.py`
- `python/sglang/jit_kernel/benchmark/bench_rmsnorm.py`
- `python/sglang/jit_kernel/benchmark/bench_rope.py`
- `python/sglang/jit_kernel/benchmark/bench_store_cache.py`
- `python/sglang/jit_kernel/norm.py`
- `python/sglang/jit_kernel/tests/test_add_constant.py`
- `python/sglang/jit_kernel/tests/test_awq_dequantize.py`
- `python/sglang/jit_kernel/tests/test_awq_marlin_moe_repack.py`
- `python/sglang/jit_kernel/tests/test_awq_marlin_repack.py`
- `python/sglang/jit_kernel/tests/test_clamp_position.py`
- `python/sglang/jit_kernel/tests/test_concat_mla.py`
- `python/sglang/jit_kernel/tests/test_custom_all_reduce.py`
- `python/sglang/jit_kernel/tests/test_cutedsl_gdn.py`
- `python/sglang/jit_kernel/tests/test_flash_attention_4.py`
- `python/sglang/jit_kernel/tests/test_fused_add_rmsnorm.py`
- `python/sglang/jit_kernel/tests/test_fused_metadata_copy.py`
- `python/sglang/jit_kernel/tests/test_fused_norm_scale_shift.py`
- `python/sglang/jit_kernel/tests/test_fused_store_index_cache.py`
- `python/sglang/jit_kernel/tests/test_fused_verify_triton_gdn.py`
- `python/sglang/jit_kernel/tests/test_gptq_marlin.py`
- `python/sglang/jit_kernel/tests/test_gptq_marlin_repack.py`
- `python/sglang/jit_kernel/tests/test_hadamard_jit.py`
- `python/sglang/jit_kernel/tests/test_moe_lora_align_block_size.py`
- `python/sglang/jit_kernel/tests/test_moe_wna16_marlin.py`
- `python/sglang/jit_kernel/tests/test_norm_jit.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/jit_kernel/tests/test_per_tensor_quant_fp8.py`
- `python/sglang/jit_kernel/tests/test_per_token_group_quant_8bit.py`
- `python/sglang/jit_kernel/tests/test_pos_enc.py`
- `python/sglang/jit_kernel/tests/test_qknorm.py`
- `python/sglang/jit_kernel/tests/test_qknorm_across_heads.py`
- `python/sglang/jit_kernel/tests/test_qwen_image_modulation.py`
- `python/sglang/jit_kernel/tests/test_renorm.py`
- `python/sglang/jit_kernel/tests/test_resolve_future_token_ids.py`
- `python/sglang/jit_kernel/tests/test_rmsnorm.py`
- `python/sglang/jit_kernel/tests/test_rope.py`
- `python/sglang/jit_kernel/tests/test_store_cache.py`
- `python/sglang/jit_kernel/tests/test_timestep_embedding.py`
- `scripts/version_branch_to_tag.sh`
- `test/run_suite.py`

## Evidence Bundle

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