Mohammad Miadh Angkad f88acf8780 [JIT Kernel] Reland NVFP4 kernels to JIT (#20012)

2026-03-07 10:31:08 +08:00

..

test_add_constant.py

[jit_kernel] Tiny unify jit_kernel tests style (#19694)

2026-03-02 21:33:59 +08:00

test_awq_dequantize.py

[jit_kernel] Tiny unify jit_kernel tests style (#19694)

2026-03-02 21:33:59 +08:00

test_awq_marlin_moe_repack.py

[jit_kernel] Tiny unify jit_kernel tests style (#19694)

2026-03-02 21:33:59 +08:00

test_awq_marlin_repack.py

[jit_kernel] Tiny unify jit_kernel tests style (#19694)

2026-03-02 21:33:59 +08:00

test_concat_mla.py

[jit_kernel] Tiny unify jit_kernel tests style (#19694)

2026-03-02 21:33:59 +08:00

test_cutedsl_gdn.py

[jit_kernel] Tiny unify jit_kernel tests style (#19694)

2026-03-02 21:33:59 +08:00

test_flash_attention_4.py

[jit_kernel] Tiny unify jit_kernel tests style (#19694)

2026-03-02 21:33:59 +08:00

test_fused_add_rmsnorm.py

[jit_kernel] Tiny unify jit_kernel tests style (#19694)

2026-03-02 21:33: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

[jit_kernel] Tiny unify jit_kernel tests style (#19694)

2026-03-02 21:33:59 +08:00

test_fused_store_index_cache.py

[jit_kernel] Tiny unify jit_kernel tests style (#19694)

2026-03-02 21:33:59 +08:00

test_fused_verify_triton_gdn.py

[GDN][Qwen3-Next][Qwen3.5] Fuse fused_gdn_gating and fused_recurrent_gated_delta_rule_update in verify_target (#19775)

2026-03-06 21:42:44 +08:00

test_gptq_marlin_repack.py

[jit_kernel] Tiny unify jit_kernel tests style (#19694)

2026-03-02 21:33:59 +08:00

test_gptq_marlin.py

[jit_kernel] Tiny unify jit_kernel tests style (#19694)

2026-03-02 21:33:59 +08:00

test_moe_wna16_marlin.py

[jit_kernel] Tiny unify jit_kernel tests style (#19694)

2026-03-02 21:33:59 +08:00

test_nvfp4_blockwise_moe.py

[JIT Kernel] Reland NVFP4 kernels to JIT (#20012)

2026-03-07 10:31:08 +08:00

test_nvfp4_gemm.py

[JIT Kernel] Reland NVFP4 kernels to JIT (#20012)

2026-03-07 10:31:08 +08:00

test_nvfp4_quant.py

[JIT Kernel] Reland NVFP4 kernels to JIT (#20012)

2026-03-07 10:31:08 +08:00

test_per_tensor_quant_fp8.py

[jit_kernel] Tiny unify jit_kernel tests style (#19694)

2026-03-02 21:33:59 +08:00

test_per_token_group_quant_8bit.py

[jit_kernel] Tiny unify jit_kernel tests style (#19694)

2026-03-02 21:33:59 +08:00

test_pos_enc.py

[jit_kernel] Tiny unify jit_kernel tests style (#19694)

2026-03-02 21:33:59 +08:00

test_qknorm_across_heads.py

[jit_kernel] Tiny unify jit_kernel tests style (#19694)

2026-03-02 21:33:59 +08:00

test_qknorm.py

[jit_kernel] Tiny unify jit_kernel tests style (#19694)

2026-03-02 21:33:59 +08:00

test_renorm.py

Migrate renorm kernels from sgl-kernel to FlashInfer JIT (#18854)

2026-03-06 22:53:28 +08:00

test_rmsnorm.py

[jit_kernel] Tiny unify jit_kernel tests style (#19694)

2026-03-02 21:33:59 +08:00

test_rope.py

[jit_kernel] Tiny unify jit_kernel tests style (#19694)

2026-03-02 21:33:59 +08:00

test_store_cache.py

[Test] Enhance JIT kvcache store kernel test coverage (#19630)

2026-03-05 16:17:15 +08:00

test_timestep_embedding.py

[jit_kernel] Tiny unify jit_kernel tests style (#19694)

2026-03-02 21:33:59 +08:00