yinfan98andSleepcoo 37c66ec856 [feat] add fa3 in sgl-kernel (#4902) ...

Co-authored-by: Sleepcoo <Sleepcoo@gmail.com>

2025-03-30 12:57:10 -07:00

..

speculative

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

2025-03-30 10:36:52 -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

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

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

test_bmm_fp8.py

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

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

test_cublas_grouped_gemm.py

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

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

test_flash_attention.py

[feat] add fa3 in sgl-kernel (#4902)

2025-03-30 12:57:10 -07:00

test_fp4_gemm.py

Support FP4 gemm (1/2) (#3899)

2025-03-24 19:50:23 -07:00

test_fp4_quantize.py

Support FP4 gemm (1/2) (#3899)

2025-03-24 19:50:23 -07:00

test_fp8_blockwise_gemm.py

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

2025-03-30 10:36:52 -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

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

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

test_lightning_attention_decode.py

[Fix] use torch.cat instead of torch.concat to prevent entering the Autograd backends. (#4466)

2025-03-16 00:02:47 -07:00

test_moe_align.py

Remove the vllm dependency from the moe_align function (#4164)

2025-03-07 22:42:16 -08: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

Add moe topk softmax templated from vllm (#4302)

2025-03-14 12:03:33 -07:00

test_norm.py

cleanup sgl-kernel kernels (#3175)

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

test_per_tensor_quant_fp8.py

[quant kernel] sgl-kernel support per_tensor_quant fp8 (#3786)

2025-03-06 18:05:43 -08:00

test_per_token_group_quant_8bit.py

[Quant Kernel] refactored per token group quant fp8 to support int8 up-to 2x faster (#4396)

2025-03-23 23:44:17 -07:00

test_per_token_quant_fp8.py

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

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

test_rotary_embedding.py

Simplify tests & Fix trtllm custom allreduce registration (#4252)

2025-03-10 01:24:22 -07:00

test_sampling.py

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

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

test_trt_allreduce.py

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

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