Richard Zou e82fbeec7b [torch.compile] Undo the fast_moe_cold_start hack in torch>=2.11 (#35475) ...

Signed-off-by: Richard Zou <zou3519@gmail.com>

2026-03-01 21:44:22 +00:00

..

configs

[Model Performance] Add Qwen3MoE tuned MoE configs for H200 (#35457)

2026-02-27 13:51:14 +08:00

oracle

[WideEP] Remove pplx all2all backend (#33724)

2026-02-26 14:30:10 -08:00

router

[Performance] Cublas Bf16 Gate with Fp32 Output (#35121)

2026-02-26 16:51:28 -08:00

runner

[torch.compile] Undo the fast_moe_cold_start hack in torch>=2.11 (#35475)

2026-03-01 21:44:22 +00:00

__init__.py

[Performance] Cublas Bf16 Gate with Fp32 Output (#35121)

2026-02-26 16:51:28 -08:00

activation.py

[Refactor] Replace activation: str with MoEActivation enum (#33843)

2026-02-11 17:29:32 -08:00

all2all_utils.py

[WideEP] Remove pplx all2all backend (#33724)

2026-02-26 14:30:10 -08:00

batched_deep_gemm_moe.py

[Refactor] Replace activation: str with MoEActivation enum (#33843)

2026-02-11 17:29:32 -08:00

config.py

[WideEP] Remove pplx all2all backend (#33724)

2026-02-26 14:30:10 -08:00

cpu_fused_moe.py

[Bugfix] fix activation in cpu_fused_moe_torch call (#34696)

2026-02-17 23:39:46 -08:00

cutlass_moe.py

[WideEP] Remove pplx all2all backend (#33724)

2026-02-26 14:30:10 -08:00

deep_gemm_moe.py

[Refactor] Replace activation: str with MoEActivation enum (#33843)

2026-02-11 17:29:32 -08:00

deep_gemm_utils.py

[Perf] Optimize deepgemm experts initialization, 3.9% TTFT improvement (#30494)

2025-12-11 17:28:34 -05:00

deepep_ht_prepare_finalize.py

[MoE Refactor] Integrate Naive Prepare Finalize into MK (#32567)

2026-01-27 01:28:02 +00:00

deepep_ll_prepare_finalize.py

[UX] Add --moe-backend arg for explicit kernel selection (#33807)

2026-02-25 17:44:44 -08:00

fallback.py

[Refactor] Replace activation: str with MoEActivation enum (#33843)

2026-02-11 17:29:32 -08:00

flashinfer_a2a_prepare_finalize.py

[MoE Refactor] Integrate Naive Prepare Finalize into MK (#32567)

2026-01-27 01:28:02 +00:00

flashinfer_cutedsl_moe.py

[Refactor] Replace activation: str with MoEActivation enum (#33843)

2026-02-11 17:29:32 -08:00

flashinfer_cutlass_moe.py

[MoE Refactor] MXFP4 Cutlass Experts to MK (#34542)

2026-02-25 17:32:39 -08:00

flashinfer_trtllm_moe.py

[UX] More descriptive reasons in is_supported_config for MoE (#34908)

2026-02-19 15:20:52 -08:00

fused_batched_moe.py

[WideEP] Remove pplx all2all backend (#33724)

2026-02-26 14:30:10 -08:00

fused_marlin_moe.py

[Model] Add NVFP4 quantization support for Step3.5-Flash (#34478)

2026-02-22 12:30:46 -07:00

fused_moe_method_base.py

[MoE Refactor] Introduce MoERunner abstraction and move execution logic from FusedMoE to DefaultMoERunner (#32344)

2026-02-10 19:51:07 -05:00

fused_moe_modular_method.py

[Bugfix] Remove assert that's no longer valid (#34443)

2026-02-12 18:18:15 -08:00

fused_moe.py

[Perf] Improve default triton fused moe configs (#34846)

2026-02-23 16:01:28 -08:00

gpt_oss_triton_kernels_moe.py

[ROCm][Quantization] GPT OSS Upstream MoE wmxfp4_afp8 with static scales (#30357)

2026-02-26 16:50:16 -06:00

layer.py

[1/N] Elastic EP Milestone 2 (#34861)

2026-02-28 04:46:42 +00:00

modular_kernel.py

[WideEP] Remove pplx all2all backend (#33724)

2026-02-26 14:30:10 -08:00

moe_align_block_size.py

[Kernel][MoE] optimize moe_align_block_size (#29642)

2025-12-07 01:58:47 -08:00

moe_permute_unpermute.py

[Refactor] Remove align block size logic in moe_permute (#33449)

2026-02-06 10:57:06 -08:00

mori_prepare_finalize.py

[MoE Refactor] Integrate Naive Prepare Finalize into MK (#32567)

2026-01-27 01:28:02 +00:00

prepare_finalize.py

[Docs] Reduce time spent generating API docs (#34255)

2026-02-11 02:56:02 -08:00

rocm_aiter_fused_moe.py

[Bugfix] Remove assert causing hipErrorStreamCaptureUnsupported (#34455)

2026-02-18 18:54:54 +00:00

routed_experts_capturer.py

[Bugfix] fix device_name for routing replay (#34336)

2026-02-26 12:18:38 +00:00

shared_fused_moe.py

[MoE Refactor] Introduce MoERunner abstraction and move execution logic from FusedMoE to DefaultMoERunner (#32344)

2026-02-10 19:51:07 -05:00

topk_weight_and_reduce.py

[WideEP] Remove pplx all2all backend (#33724)

2026-02-26 14:30:10 -08:00

triton_cutlass_moe.py

[Refactor] Replace activation: str with MoEActivation enum (#33843)

2026-02-11 17:29:32 -08:00

triton_deep_gemm_moe.py

[Refactor] Replace activation: str with MoEActivation enum (#33843)

2026-02-11 17:29:32 -08:00

trtllm_moe.py

[MoE Refactor] MXFP4 Cutlass Experts to MK (#34542)

2026-02-25 17:32:39 -08:00

unquantized_fused_moe_method.py

[MoE Refactor] Introduce MoERunner abstraction and move execution logic from FusedMoE to DefaultMoERunner (#32344)

2026-02-10 19:51:07 -05:00

utils.py

[MoE Refactor] MXFP4 Cutlass Experts to MK (#34542)

2026-02-25 17:32:39 -08:00

xpu_fused_moe.py

[Refactor] Replace activation: str with MoEActivation enum (#33843)

2026-02-11 17:29:32 -08:00

zero_expert_fused_moe.py

[MoE Refactor][5/N] Isolate zero expert to LongCatFlash (#28891)

2025-12-20 18:22:04 +00:00