Mohammad Miadh Angkad 2bdd89a6cd [Kernel Slimming] Migrate NVFP4 kernels to JIT (#19437)

2026-03-05 15:22:28 +08:00

..

bench_awq_dequantize.py

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

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

bench_awq_marlin_moe_repack.py

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

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

bench_awq_marlin_repack.py

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

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

bench_concat_mla.py

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

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

bench_fused_add_rmsnorm.py

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

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

bench_fused_norm_scale_shift.py

[diffusion] kernel: gated residual layernorm scale shift and layernorm scale shift kernel fusion for Qwen-Image, WAN and HunyuanVideo (#14717)

2026-02-04 13:46:20 +08:00

bench_gptq_marlin_repack.py

[Kernel Slimming] Migrate GPTQ-Marlin repack kernel to JIT (#18543)

2026-02-13 22:29:22 +08:00

bench_gptq_marlin.py

[Kernel] Migrate GPTQ-Marlin GEMM kernel to JIT (#18067)

2026-02-06 08:31:42 +08:00

bench_hicache.py

[Refactor] Fix test and clean up hicache code (#18555)

2026-02-18 14:37:46 +08:00

bench_moe_wna16_marlin.py

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

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

bench_nvfp4_blockwise_moe.py

[Kernel Slimming] Migrate NVFP4 kernels to JIT (#19437)

2026-03-05 15:22:28 +08:00

bench_nvfp4_quant.py

[Kernel Slimming] Migrate NVFP4 kernels to JIT (#19437)

2026-03-05 15:22:28 +08:00

bench_nvfp4_scaled_mm.py

[Kernel Slimming] Migrate NVFP4 kernels to JIT (#19437)

2026-03-05 15:22:28 +08:00

bench_per_tensor_quant_fp8.py

add hicache jit test (#17847)

2026-02-06 16:54:33 +08:00

bench_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

bench_qknorm_across_heads.py

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

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

bench_qknorm.py

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

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

bench_rmsnorm.py

add hicache jit test (#17847)

2026-02-06 16:54:33 +08:00

bench_rope.py

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

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

bench_store_cache.py

[AMD] Enable ROCm kvcache JIT path and add AMD CI coverage. (#18992)

2026-02-25 14:15:05 +08:00

utils.py

add hicache jit test (#17847)

2026-02-06 16:54:33 +08:00