Xiaoyu Zhang 515ef4facb Fuse routed scaling factor in topk_reduce kernel (#6220)

2025-06-07 11:06:50 -07:00

..

configs

Add triton version as a fused_moe_triton config search key to avoid performace decrease in different Triton version (#5955)

2025-06-07 02:43:50 -07:00

__init__.py

Reorg moe code (#2563)

2024-12-24 01:10:22 +08:00

fused_moe.py

Fuse routed scaling factor in topk_reduce kernel (#6220)

2025-06-07 11:06:50 -07:00

layer.py

Fuse routed scaling factor in topk_reduce kernel (#6220)

2025-06-07 11:06:50 -07:00