zyl_keep_movingandXiaoyu Zhang a9ce1623cd [kernel][moe] add moe topk fast (#13969) ...

Co-authored-by: Xiaoyu Zhang <35585791+BBuf@users.noreply.github.com>

2025-12-14 22:26:40 +08:00

..

spatial

…

speculative

…

conftest.py

…

test_activation.py

…

test_apply_token_bitmask_inplace.py

…

test_awq_dequant.py

Fix AWQ Dequant and Weight Loading of deepseek v2 (#6842)

2025-06-17 13:45:10 -07:00

test_bmm_fp8.py

…

test_causal_conv1d.py

Add mamba kernel (#10234)

2025-09-09 12:58:43 -07:00

test_copy.py

…

test_custom_allreduce.py

sgl-kernel transfer custom allreduce from trt kernel to vllm kernel (#5079)

2025-04-05 14:23:20 -07:00

test_cutlass_mla.py

Disable kernel cutlass_mla_decode on SM103 (#10058)

2025-09-06 01:35:18 -07:00

test_cutlass_w4a8_moe_mm.py

…

test_dsv3_fused_a_gemm.py

…

test_dsv3_router_gemm.py

…

test_es_fp8_blockwise_moe.py

…

test_es_mxfp8_blockscaled_moe.py

…

test_flash_attention_4.py

…

test_flash_attention.py

…

test_flashmla.py

…

test_fp4_gemm.py

[Misc] Use pytest.mark.skipif in sgl-kernel test (#5137)

2025-04-07 21:35:14 -07:00

test_fp4_quantize.py

…

test_fp8_blockwise_gemm.py

…

test_fp8_blockwise_moe.py

…

test_fp8_gemm.py

…

test_fused_qk_norm_rope.py

…

test_gguf.py

…

test_gptq_kernel.py

…

test_hadamard.py

…

test_int8_gemm.py

…

test_kimi_k2_moe_fused_gate.py

…

test_kvcacheio.py

…

test_marlin_gemm.py

…

test_marlin_repack.py

[2/n]decouple quantization implementation from vLLM dependency (#8112)

2025-08-14 03:19:03 -07:00

test_merge_state_v2.py

…

test_merge_state.py

Add main for merge state tests (#6492)

2025-05-21 21:56:25 -07:00

test_moe_align.py

…

test_moe_fused_gate.py

[2/2] Fuse routed scaling factor into select_experts (#8690)

2025-08-20 15:10:16 -07:00

test_moe_topk_sigmoid.py

…

test_moe_topk_softmax.py

…

test_mscclpp.py

…

test_norm.py

…

test_per_tensor_quant_fp8.py

…

test_per_token_group_quant_8bit.py

…

test_per_token_quant_fp8.py

…

test_qserve_w4a8_per_chn_gemm.py

…

test_qserve_w4a8_per_group_gemm.py

…

test_rotary_embedding.py

[1/2] Add rope kernel in sgl-kernel (#14334)

2025-12-04 16:45:44 +08:00

test_sampling.py

…

test_sparse_flash_attn.py

…

test_topk.py

…

test_torch_defaults_reset.py

…

utils.py

…