Amazon SageMaker AI 上的 Mixture-of-Experts 内核工程优化
AWS 分享了在 Amazon SageMaker AI 上针对 fused MoE 模型的内核优化:通过 Swap-AB 让专家维度 N=8192 进入 WGMMA 首个操作数位置,在 H200 上带来 2–4% 端到端输出吞吐提升,该优化已合入 vLLM 0.24.0 或更高版本。此外,在 Hopper 上对 FP8 张量量化模型,Triton 后端在低并发 MoE 负载下持续优于默认的 FLASHINFER_CUTLASS 路径。

AWS 分享了在 Amazon SageMaker AI 上针对 fused MoE 模型的内核优化:通过 Swap-AB 让专家维度 N=8192 进入 WGMMA 首个操作数位置,在 H200 上带来 2–4% 端到端输出吞吐提升,该优化已合入 vLLM 0.24.0 或更高版本。此外,在 Hopper 上对 FP8 张量量化模型,Triton 后端在低并发 MoE 负载下持续优于默认的 FLASHINFER CUTLASS 路径。
匹配字段摘要、正文




