Baizhou Zhang 282eb59ff3 Add bf16 output option for dsv3_router_gemm kernel (#7999)

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

..

bench_awq_dequant.py

…

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_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

…

bench_int8_gemm.py

…

bench_lightning_attention_decode.py

…

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

…

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

…

bench_moe_topk_softmax.py

[optimize] fuse renormalize into moe_topk_softmax (#7744)

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

bench_per_tensor_quant_fp8.py

…

bench_per_token_group_quant_8bit.py

…

bench_per_token_quant_fp8.py

…

bench_qserve_w4a8_gemm.py

…