Ke Bao a3398d8478 Optimize moe align block size kernel (#7794)

2025-07-07 09:20:30 +08:00

..

cutlass_moe/w4a8

[1/n]: add cutlass W4A8 moe kernel for hopper architecture (#7772)

2025-07-04 20:50:12 -07:00

marlin_moe_wna16

[1/n] apply wna16marlin kernel in moe weight only quantization (#7683)

2025-07-01 23:21:25 -07:00

cutlass_moe_helper.cu

[1/2] Add FP8 Blockscale MoE CUTLASS kernel for Blackwell (#5281)

2025-04-22 22:28:20 -07:00

ep_moe_reorder_kernel.cu

[EP] Add cuda kernel for moe_ep_post_reorder (#6837)

2025-06-05 00:33:47 -07:00

ep_moe_silu_and_mul_kernel.cu

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

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

fp8_blockwise_moe_kernel.cu

Optimize Hopper CUTLASS FP8 Blockwise Grouped GEMM Kernel in Small K Scenario (#7782)

2025-07-04 22:25:49 -07:00

moe_align_kernel.cu

Optimize moe align block size kernel (#7794)

2025-07-07 09:20:30 +08:00

moe_fused_gate.cu

Set num_fused_shared_experts as num_shared_experts when shared_experts fusion is not disabled (#6736)

2025-06-04 15:53:22 -07:00

moe_topk_softmax_kernels.cu

[optimize] fuse renormalize into moe_topk_softmax (#7744)

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

nvfp4_blockwise_moe.cu

[1/2] Add Kernel support for Cutlass based Fused FP4 MoE (#6093)

2025-06-02 13:48:03 -07:00

prepare_moe_input.cu

fix: fix apply_shuffle_mul_sum (#7444)

2025-07-04 23:23:30 -07:00