Lucia Fang 7e082bc14e

Support DeepEP for Kimi-k2-thinking through enabling gemm selection for compressed-tensor marlin wna16 (#28574) ...

Signed-off-by: Lu Fang <fanglu@fb.com>

2025-11-12 21:40:45 -08:00

..

configs

[Hardware][AMD][Model] Add Triton MoE tuning support and optimized configs for Qwen3 omni for MI308X (#28373)

2025-11-10 04:53:40 +00:00

__init__.py

[Kernels] Split up fused_moe/layer.py, isolate more modular kernel code (#28064)

2025-11-11 07:29:02 -07:00

all2all_utils.py

[Kernels] Split up fused_moe/layer.py, isolate more modular kernel code (#28064)

2025-11-11 07:29:02 -07:00

batched_deep_gemm_moe.py

[Performance][B200] Fix deepgemm prologue (#27897)

2025-11-12 13:13:03 -08:00

batched_triton_or_deep_gemm_moe.py

…

config.py

[Performance] Support FP8 flashinfer TRTLLM MOE on Qwen3 and Qwen-3next (#27492)

2025-11-10 12:34:57 -05:00

cpu_fused_moe.py

[CPU]Improve cpu fused moe perf (#27244)

2025-11-06 11:04:18 +00:00

cutlass_moe.py

…

deep_gemm_moe.py

[BugFix] Fix DeepGEMM over-allocating workspace (#28254)

2025-11-10 17:01:17 -05:00

deep_gemm_utils.py

…

deepep_ht_prepare_finalize.py

[Bug] Fix DBO IMA issue for DeepEPHT (#27666)

2025-10-29 16:28:27 -04:00

deepep_ll_prepare_finalize.py

[Performance][B200] Fix deepgemm prologue (#27897)

2025-11-12 13:13:03 -08:00

flashinfer_cutlass_moe.py

Flashinfer_CUTLASS_MOE fuses quantization for TP (#27223)

2025-10-31 17:54:29 +00:00

flashinfer_cutlass_prepare_finalize.py

[flashinfer] fix FI all2all with FI cutlass moe (#28166)

2025-11-06 05:52:16 +00:00

flashinfer_trtllm_moe.py

[Performance] Support FP8 flashinfer TRTLLM MOE on Qwen3 and Qwen-3next (#27492)

2025-11-10 12:34:57 -05:00

fused_batched_moe.py

…

fused_marlin_moe.py

Support DeepEP for Kimi-k2-thinking through enabling gemm selection for compressed-tensor marlin wna16 (#28574)

2025-11-12 21:40:45 -08:00

fused_moe_method_base.py

[Kernels] Split up fused_moe/layer.py, isolate more modular kernel code (#28064)

2025-11-11 07:29:02 -07:00

fused_moe_modular_method.py

[Kernels] Split up fused_moe/layer.py, isolate more modular kernel code (#28064)

2025-11-11 07:29:02 -07:00

fused_moe.py

[RFC][ROCm][AITER] Keep all AITER kernels in _aiter_ops class like _custom_ops and _ipex_ops (#24490)

2025-11-10 08:20:53 -08:00

gpt_oss_triton_kernels_moe.py

…

layer.py

[MoE][Kernel][Perf] Improve Shared Expert Stream Overlap (#28406)

2025-11-12 23:37:24 +00:00

modular_kernel.py

[Performance][B200] Fix deepgemm prologue (#27897)

2025-11-12 13:13:03 -08:00

moe_align_block_size.py

…

moe_pallas.py

…

moe_permute_unpermute.py

…

moe_torch_iterative.py

…

pplx_prepare_finalize.py

…

prepare_finalize.py

…

rocm_aiter_fused_moe.py

[RFC][ROCm][AITER] Keep all AITER kernels in _aiter_ops class like _custom_ops and _ipex_ops (#24490)

2025-11-10 08:20:53 -08:00

routing_simulator.py

[Bugfix] Fix MoE Routing Simulation (#28002)

2025-11-03 22:26:49 +00:00

shared_fused_moe.py

[Bugfix] Disable shared expert overlap if Marlin MoE is used (#28410)

2025-11-11 23:16:12 +00:00

topk_weight_and_reduce.py

…

triton_deep_gemm_moe.py

…

trtllm_moe.py

[BugFix][Performance] Restore flashinfer autotuning for all scenarios (#27904)

2025-11-04 15:56:21 +08:00

unquantized_fused_moe_method.py

[Kernels] Split up fused_moe/layer.py, isolate more modular kernel code (#28064)

2025-11-11 07:29:02 -07:00

utils.py

…