Qingquan Song 61e4433caf Add moe topk softmax templated from vllm (#4302)

2025-03-14 12:03:33 -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

Add awq dequantize kernel to sgl with 1x to 3x speedup (#4104)

2025-03-12 00:10:02 -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] Apply Cublas Grouped Gemm kernel (#3629)

2025-02-18 15:18:31 +08:00

test_deep_gemm.py

DeepGemm integrate to sgl-kernel (#4165)

2025-03-10 00:35:07 -07:00

test_fp8_blockwise_gemm.py

support blockwise fp8 matmul kernel (#3267)

2025-02-13 01:49:33 +08:00

test_fp8_gemm.py

support w8a8 fp8 kernel with CUTLASS (#3047)

2025-01-26 15:46:51 +08:00

test_int8_gemm.py

Support sm90 Int8 gemm (#3035)

2025-01-21 22:21:54 +08:00

test_lightning_attention_decode.py

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

2025-01-23 21:13:55 +08: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_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_fp8.py

fix per_token_group_quant_fp8 illegal memory when num_groups % 16 != 0 (#4231)

2025-03-10 01:42:58 -07:00

test_per_token_quant_fp8.py

fix accuracy issue (#4376)

2025-03-13 02:06:22 -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_speculative_sampling.py

support speculative decoding kernel in sgl-kernel (#3373)

2025-02-07 20:29:51 +08:00

test_trt_allreduce.py

Rename files in sgl kernel to avoid nested folder structure (#4213)

2025-03-08 22:54:51 -08:00