跳到正文
AWS Machine Learning Blog· Mona Mona·· 4 小时前AI 评分32

Amazon SageMaker AI 上 MoE 模型的 Kernel 工程优化

Mixture-of-Experts kernel engineering on Amazon SageMaker AI

AI 导读

AWS 在 Amazon SageMaker AI 上针对 Mixture-of-Experts 架构进行 kernel 工程优化,通过 Swap-AB 让 WGMMA 指令矩阵对齐 Hopper Tensor Core 布局,并在 H200 上为大规�kgate MoE 模型带来 2–4% 的端到端吞吐提升,相关优化已合入 vLLM 0.24.0。

来源:AWS Machine Learning Blog · aws.amazon.com