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

2025-04-05 14:23:20 -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_custom_allreduce.py

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

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

test_flash_attention.py

[Fix] fix fa3 build at cu118 (#5036)

2025-04-03 11:52:35 -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

Add DeepSeek V3/R1 shared experts fusion (#4918)

2025-04-04 01:59:29 -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

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

[sgl-kernel] per token group quant support COLUMN MAJOR (#4817)

2025-04-02 18:29:59 -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