Yuan Luo 121f92c583 Add main for merge state tests (#6492) ...

Co-authored-by: luoyuan.luo <luoyuan.luo@antgroup.com>

2025-05-21 21:56:25 -07:00

..

speculative

[Misc] Clean sgl-kernel test (#5216)

2025-04-10 11:28:41 -07:00

test_activation.py

sync flashinfer and update sgl-kernel tests (#3081)

2025-01-23 21:13:55 +08:00

test_apply_token_bitmask_inplace.py

fix sgl-kernel unit tests (#5666)

2025-04-23 01:18:30 -07:00

test_awq_dequant.py

[Misc] clean up vllm in sgl-kernel test (#5189)

2025-04-09 01:22:13 -07:00

test_bmm_fp8.py

feat: integrate bmm_fp8 kernel into sgl-kernel (#3056)

2025-01-23 00:39:38 +08:00

test_custom_allreduce.py

sgl-kernel transfer custom allreduce from trt kernel to vllm kernel (#5079)

2025-04-05 14:23:20 -07:00

test_cutlass_mla.py

Cutlass MLA: Disable split kv due to https://github.com/NVIDIA/cutlass/issues/2274 (#6101)

2025-05-08 18:44:30 -07:00

test_flash_attention.py

Add typo checker in pre-commit (#6179)

2025-05-11 12:55:00 +08:00

test_fp4_gemm.py

[Misc] Use pytest.mark.skipif in sgl-kernel test (#5137)

2025-04-07 21:35:14 -07:00

test_fp4_quantize.py

[Misc] Use pytest.mark.skipif in sgl-kernel test (#5137)

2025-04-07 21:35:14 -07:00

test_fp8_blockwise_gemm.py

[Misc] Clean sgl-kernel test (#5216)

2025-04-10 11:28:41 -07:00

test_fp8_blockwise_moe.py

[2/2] Add python wrapper for CUTLASS FP8 Blockscale MoE Kernel. (#5694)

2025-05-16 13:14:07 -07:00

test_fp8_gemm.py

[Feature] use pytest for sgl-kernel (#4896)

2025-03-30 10:36:52 -07:00

test_int8_gemm.py

[Misc] Clean sgl-kernel test (#5216)

2025-04-10 11:28:41 -07:00

test_lightning_attention_decode.py

[Misc] Clean sgl-kernel test (#5216)

2025-04-10 11:28:41 -07:00

test_merge_state_v2.py

Add main for merge state tests (#6492)

2025-05-21 21:56:25 -07:00

test_merge_state.py

Add main for merge state tests (#6492)

2025-05-21 21:56:25 -07:00

test_moe_align.py

reduce moe_align_block_size_kernel small batch mode overhead (#5086)

2025-04-09 17:59:35 -07:00

test_moe_fused_gate.py

fix sgl-kernel unit tests (#5666)

2025-04-23 01:18:30 -07:00

test_moe_topk_softmax.py

[Misc] Clean sgl-kernel test (#5216)

2025-04-10 11:28:41 -07:00

test_norm.py

cleanup sgl-kernel kernels (#3175)

2025-01-27 19:11:01 +08:00

test_per_tensor_quant_fp8.py

update variable naming and comments for rocm (#5299)

2025-04-11 23:15:05 -07:00

test_per_token_group_quant_8bit.py

Add typo checker in pre-commit (#6179)

2025-05-11 12:55:00 +08:00

test_per_token_quant_fp8.py

update variable naming and comments for rocm (#5299)

2025-04-11 23:15:05 -07:00

test_qserve_w4a8_per_chn_gemm.py

[1/2] Support Qserve (#6457)

2025-05-21 19:48:59 -07:00

test_qserve_w4a8_per_group_gemm.py

[1/2] Support Qserve (#6457)

2025-05-21 19:48:59 -07:00

test_rotary_embedding.py

[Misc] Clean sgl-kernel test (#5216)

2025-04-10 11:28:41 -07:00

test_sampling.py

[Feat] Update sgl-kernel flashinfer to latest main version (#5500)

2025-04-17 12:43:23 -07:00

test_sparse_flash_attn.py

[Feat] QWen-1M context support[1/2]: Update block sparse attention backend utils kernel (#5847)

2025-04-28 11:03:17 -07:00