Yuan Luo 7c9e8e2def [Re-land][jit kernel] Support per_token_group_quant_8bit jit kernel (#19140) ...

Co-authored-by: luoyuan.luo <luoyuan.luo@antgroup.com>
Co-authored-by: Mohammad Miadh Angkad <mangkad.bsdsba2027@aim.edu>

2026-02-26 09:53:57 +08:00

..

test_add_constant.py

…

test_awq_dequantize.py

[Kernel Slimming] Migrate AWQ marlin repack kernel to JIT (#18949)

2026-02-23 22:05:27 +08:00

test_awq_marlin_moe_repack.py

[Kernel Slimming] Migrate AWQ marlin repack kernel to JIT (#18949)

2026-02-23 22:05:27 +08:00

test_awq_marlin_repack.py

[Kernel Slimming] Migrate AWQ marlin repack kernel to JIT (#18949)

2026-02-23 22:05:27 +08:00

test_concat_mla.py

[Move sgl-kernel Kernel to JIT] Add JIT concat MLA kernels (#17889)

2026-02-03 10:49:17 +08:00

test_cutedsl_gdn.py

[CI] Skip cutedsl gdn performance test in jit_kernel ci (#18783)

2026-02-13 15:49:30 +08:00

test_flash_attention_4.py

Move fa4 from sgl-kernel to jit kernel (#17353)

2026-01-24 15:25:03 +08:00

test_fused_add_rmsnorm.py

[JIT Kernel]Support fused_add_rmsnorm in JIT Kernel (#17677)

2026-01-29 09:29:59 +08:00

test_fused_metadata_copy.py

Kernel: optimize decoding metadata in NSA multi-spec backend with fused kernels (#17554)

2026-02-14 16:40:15 +08:00

test_fused_norm_scale_shift.py

[diffusion] fix: fix torch.compile graph break caused by torch._dynamo.disable (#18336)

2026-02-06 14:48:09 +08:00

test_gptq_marlin_repack.py

[Kernel Slimming] Remove sgl-kernel AOT marlin kernels (#19241)

2026-02-25 10:08:22 +08:00

test_gptq_marlin.py

[Kernel Slimming] Remove sgl-kernel AOT marlin kernels (#19241)

2026-02-25 10:08:22 +08:00

test_moe_wna16_marlin.py

[Kernel Slimming] Migrate marlin moe kernel to JIT (#19181)

2026-02-26 09:05:13 +08:00

test_per_tensor_quant_fp8.py

[Refactor] Clean up JIT kernel utilites (#16884)

2026-01-13 17:54:16 +08:00

test_per_token_group_quant_8bit.py

[Re-land][jit kernel] Support per_token_group_quant_8bit jit kernel (#19140)

2026-02-26 09:53:57 +08:00

test_pos_enc.py

[Kernel] Add JIT rotary_embedding_kernel (#17934)

2026-02-13 12:41:25 +08:00

test_qknorm_across_heads.py

[Diffsuion & JIT_kernel] QKNorm cross heads kernel (#18073)

2026-02-03 10:03:17 +08:00

test_qknorm.py

[JIT kernel] hd=512,1024 in JIT QK norm (cta based) (#17515)

2026-02-16 16:07:24 +08:00

test_rmsnorm.py

[Refactor] Clean up JIT kernel utilites (#16884)

2026-01-13 17:54:16 +08:00

test_rope.py

[Feature] rewrite rope kernel; remove flashinfer dependencies (#18844)

2026-02-21 21:32:40 +08:00

test_store_cache.py

[Feature] Support JIT set kv cache (#16273)

2026-01-10 17:34:09 -08:00

test_timestep_embedding.py

[Diffusion] Add diffusion time embedding to jit kernel (#17658)

2026-01-24 14:27:08 +08:00