Elfie Guo 5c9c275bc8 Use FlashInfer FP4 gemm. (#8241)

2025-07-27 01:05:22 -07:00

..

bench_activation.py

[AMD] Add silu_and_mul, gelu_and_mul, gelu_tanh_and_mul, and gelu_quick kernels for AMD GPUs (#7135)

2025-07-24 23:44:28 -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 bf16 output option for dsv3_router_gemm kernel (#7999)

2025-07-20 09:49:37 +08:00

bench_fp4_gemm.py

Use FlashInfer FP4 gemm. (#8241)

2025-07-27 01:05:22 -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_blockwise_group_gemm.py

Add CUTLASS FP8 Blockscale MoE kernel for Hopper architecture (#7278)

2025-07-02 23:27:03 -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

[fix] benchmark : routed_scaling_factor is None (#8059)

2025-07-22 08:55:35 -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

[optimize] fuse renormalize into moe_topk_softmax (#7744)

2025-07-03 12:42:44 -07:00

bench_nvfp4_scaled_gemm.py

Add nvfp4 scaled mm benchmark. (#8401)

2025-07-26 23:18:04 -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

Fix bench script making input data on L2 cache (#7739)

2025-07-27 00:30:24 -07: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