danisereb dea63512bb Add config file for fused MoE for Nemotron (TP4, B200) (#34411) ...

Signed-off-by: Daniel Serebrenik <daserebrenik@nvidia.com>

2026-02-12 06:09:55 -08:00

..

configs

Add config file for fused MoE for Nemotron (TP4, B200) (#34411)

2026-02-12 06:09:55 -08:00

oracle

[NVIDIA][test] Tests for flashinfer TRTLLM BF16 MoE (#33715)

2026-02-11 12:38:11 +00:00

router

Fix RoutingMethodType logic (#33919)

2026-02-06 14:03:34 -08:00

runner

[BugFix] Fix DP chunking (#34379)

2026-02-12 06:44:03 +00:00

__init__.py

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

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

activation.py

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

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

all2all_utils.py

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

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

batched_deep_gemm_moe.py

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

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

config.py

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

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

cpu_fused_moe.py

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

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

cutlass_moe.py

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

2026-02-11 17:29:32 -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

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

2026-01-27 01:28:02 +00: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

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

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

flashinfer_trtllm_moe.py

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

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

fused_batched_moe.py

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

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

fused_marlin_moe.py

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

2026-02-11 17:29:32 -08: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

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

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

fused_moe.py

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

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

gpt_oss_triton_kernels_moe.py

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

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

layer.py

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

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

modular_kernel.py

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

2026-02-11 17:29:32 -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

pplx_prepare_finalize.py

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

2026-02-11 02:56:02 -08: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

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

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

routed_experts_capturer.py

[Bugfix] Refactor to support DP parallel in R3 (#32306)

2026-01-16 15:13:58 +08: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

Update Optional[x] -> x | None and Union[x, y] to x | y (#26633)

2025-10-12 09:51:31 -07: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

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

2026-02-11 17:29:32 -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

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

2026-02-11 17:29:32 -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