Lianmin Zheng 76e4a8662c Replace clamp_position with JIT kernel + platform dispatch (#20999) ...

Co-authored-by: Claude Opus 4.6 (1M context) <noreply@anthropic.com>

2026-03-21 21:26:26 -07: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

Release sglang kernel 0.4.0 (#20440)

2026-03-16 20:34:58 +08:00

test_awq_marlin_repack.py

Release sglang kernel 0.4.0 (#20440)

2026-03-16 20:34:58 +08:00

test_clamp_position.py

Replace clamp_position with JIT kernel + platform dispatch (#20999)

2026-03-21 21:26:26 -07:00

test_concat_mla.py

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

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

test_custom_all_reduce.py

[JIT Kernel][Feature] Support JIT custom all reduce (rewrite as v2) (#19880)

2026-03-20 18:24:07 +08:00

test_cutedsl_gdn.py

[jit_kernel] Temporarily Skip Flaky JIT Kernel GDN Test and Add PR Label (#20436)

2026-03-13 09:34:22 +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

[4/n jit_kernel restruct] speed up CI tests and add benchmark workflow (#20268)

2026-03-10 21:37:41 +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_hadamard_jit.py

[JIT Kernel] Add hadamard kernel test and benchmark (#20030)

2026-03-17 23:16:35 -07:00

test_moe_lora_align_block_size.py

[LoRA][I] Add MOE LoRA JIT alignment kernel and tests (#19710)

2026-03-12 12:23:46 -07:00

test_moe_wna16_marlin.py

Release sglang kernel 0.4.0 (#20440)

2026-03-16 20:34:58 +08:00

test_norm_jit.py

Migrate norm kernels to FlashInfer JIT implementation (#18871)

2026-03-10 14:56:07 +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

[2/n jit_kernel restruct] unify rotary embedding entrypoints under rope.py (#20247)

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

test_qknorm_across_heads.py

[4/n jit_kernel restruct] speed up CI tests and add benchmark workflow (#20268)

2026-03-10 21:37:41 +08:00

test_qknorm.py

[4/n jit_kernel restruct] speed up CI tests and add benchmark workflow (#20268)

2026-03-10 21:37:41 +08:00

test_qwen_image_modulation.py

[Diffusion] Opt qwen-image-edit with fuse_residual_layernorm_scale_shift_gate_select01_kernel (#20395)

2026-03-13 13:15:22 +08:00

test_renorm.py

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

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

test_resolve_future_token_ids.py

Replace _resolve_future_token_ids with JIT kernel + platform dispatch (#20976)

2026-03-20 01:47:03 -07:00

test_rmsnorm.py

[4/n jit_kernel restruct] speed up CI tests and add benchmark workflow (#20268)

2026-03-10 21:37:41 +08:00

test_rope.py

[4/n jit_kernel restruct] speed up CI tests and add benchmark workflow (#20268)

2026-03-10 21:37:41 +08:00

test_store_cache.py

[4/n jit_kernel restruct] speed up CI tests and add benchmark workflow (#20268)

2026-03-10 21:37:41 +08:00

test_timestep_embedding.py

[4/n jit_kernel restruct] speed up CI tests and add benchmark workflow (#20268)

2026-03-10 21:37:41 +08:00