Baizhou Zhang 81c891111f Add test for flash_attn_varlen_func kernel (#5484)

2025-04-17 01:42:56 -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_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

BLackwell cutlass mla: Add check for bad page size/block num combinations (#5431)

2025-04-15 14:07:42 -07:00

test_flash_attention.py

Add test for flash_attn_varlen_func kernel (#5484)

2025-04-17 01:42:56 -07: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_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

kernel: support slightly faster merge_state_v2 cuda kernel (#5381)

2025-04-14 21:28:23 -07:00

test_merge_state.py

feat: adapt merge_state (#5337)

2025-04-12 21:14:04 -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

Add deepseek style fused moe group gate selection kernel (#4530)

2025-03-29 11:51:45 -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

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

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

test_per_token_quant_fp8.py

update variable naming and comments for rocm (#5299)

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

test_rotary_embedding.py

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

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

test_sampling.py

feat: integrate sampling kernels into sgl-kernel (#3086)

2025-01-24 01:54:47 +08:00

test_sparse_flash_attn.py

[Feat] Add sparse attn to sgl-kernel (#5327)

2025-04-12 11:36:36 -07:00