Baizhou Zhang 7248272ccc Add dsv3 router gemm kernel (#7627)

2025-06-29 23:31:55 -07:00

..

bench_awq_dequant.py

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

2025-03-12 00:10:02 -07:00

bench_cutlass_mla.py

[fix] fix cutlass_mla_backend with cuda_graph and add sm_scale for sgl-kernel cutlass_mla (#7184)

2025-06-14 12:45:41 -07:00

bench_dsv3_fused_a_gemm.py

Add dsv3 fused a gemm to sgl-kernel (#7630)

2025-06-29 02:52:24 -07:00

bench_dsv3_router_gemm.py

Add dsv3 router gemm kernel (#7627)

2025-06-29 23:31:55 -07:00

bench_fp8_blockwise_gemm.py

refactor apply_w8a8_block_fp8_linear in fp (#6545)

2025-05-29 00:15:11 -07:00

bench_fp8_gemm.py

support w8a8 fp8 kernel with CUTLASS (#3047)

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

bench_int8_gemm.py

Add shapes for int8 gemm benchmark (#3093)

2025-01-24 12:27:30 +08:00

bench_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

bench_moe_align_block_size.py

Fuse sorted_token_ids padding to moe_align_block_size kernel (#7437)

2025-06-24 17:44:27 -07:00

bench_moe_ep_post_reorder.py

feat: integrate deepgemm into EPMoE (#6821)

2025-06-23 01:38:58 -07:00

bench_moe_ep_pre_reorder.py

fix ep_moe_reorder kernel bugs (#6858)

2025-06-04 19:13:59 +08:00

bench_moe_fused_gate.py

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

2025-03-29 11:51:45 -07:00

bench_moe_silu_and_mul.py

[sgl-kernel] Add cuda kernel for moe_ep_silu_and_mul (#6919)

2025-06-11 20:43:08 -07:00

bench_moe_topk_softmax.py

Add moe topk softmax templated from vllm (#4302)

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

bench_per_tensor_quant_fp8.py

update variable naming and comments for rocm (#5299)

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

bench_per_token_group_quant_8bit.py

Add typo checker in pre-commit (#6179)

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

bench_per_token_quant_fp8.py

update variable naming and comments for rocm (#5299)

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

bench_qserve_w4a8_gemm.py

[1/2] Support Qserve (#6457)

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