Taoyu Zhu 70485a11bd [ROCM] Optimize the fused_topk_bias to use aiter instead of fallback torch ops. (#36253) ...

Signed-off-by: zhutaoyu <zhutaoyu97@gmail.com>

2026-03-09 11:30:35 -05:00

..

__init__.py

[MoE Refactor] Separate Router into OO Classes (#30623)

2026-01-18 11:40:49 -05:00

base_router.py

[MoE Refactor] Create MK for TRTLLM Kernels (#32564)

2026-03-03 10:39:50 -08:00

custom_routing_router.py

[MoE Refactor] Move select_experts from FusedMoEQuantMethod -> FusedMoE (#31996)

2026-01-22 18:21:35 -05:00

fused_moe_router.py

[MoE Refactor] Separate Router into OO Classes (#30623)

2026-01-18 11:40:49 -05:00

fused_topk_bias_router.py

[ROCM] Optimize the fused_topk_bias to use aiter instead of fallback torch ops. (#36253)

2026-03-09 11:30:35 -05:00

fused_topk_router.py

[Bugfix][MoE Kernel] Fix incorrect routing selection for models without expert groups (e.g., MiniMax-M2.1) (#34673)

2026-02-18 13:03:24 -08:00

gate_linear.py

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

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

grouped_topk_router.py

[Bugfix][MoE Kernel] Fix incorrect routing selection for models without expert groups (e.g., MiniMax-M2.1) (#34673)

2026-02-18 13:03:24 -08:00

router_factory.py

[Misc] Fix typos in comments: explict→explicit, paramaters→parameters (#35648)

2026-03-02 21:59:14 -08:00

routing_simulator_router.py

[MoE Refactor] Separate Router into OO Classes (#30623)

2026-01-18 11:40:49 -05:00