AWS Machine Learning Blog· Mona Mona·· 2 小时前
Amazon SageMaker AI 上的 MoE 内核工程优化实践
Mixture-of-Experts kernel engineering on Amazon SageMaker AI
AI 导读
针对 MoE 模型在解码阶段 M 维过小导致 Hopper WGMMA 利用率低的问题,Swap-AB 优化将专家维度 N=8192 提升为首操作数,在 H200 上为大规模 MoE 模型带来 2–4% 端到端输出吞吐提升,该优化已合入 vLLM 0.24.0 及更高版本。
来源:AWS Machine Learning Blog · aws.amazon.com