amirkl94 03ee48111d

Feature: Support Relu2 in FusedMoE fp8 cutlass path (#27261)

2025-11-16 13:39:44 -05:00

..

configs

[Kernel][Moe Configs] llama4 maverick fp8 moe config tp8 on mi325 (#28709)

2025-11-15 01:10:48 -08: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][DeepGEMM] Estimate expected_m (#28694)

2025-11-15 13:52:14 +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

Feature: Support Relu2 in FusedMoE fp8 cutlass path (#27261)

2025-11-16 13:39:44 -05:00

flashinfer_cutlass_prepare_finalize.py

[Kernels] Enable FlashInfer FP8 Blockscale on SM90 (for TEP DSR1) (#27134)

2025-11-14 08:02:44 -08: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

[Bugfix] Fix incorrect use of hidden_states for shared_experts due to do_naive_dispatch_combine (#28740)

2025-11-14 14:13:46 -08:00

modular_kernel.py

[BugFix] Fix misprint introduced by modular_kernel refactoring. (#28728)

2025-11-14 10:58:18 -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

…