Skip to content
Read the original: AWS Machine Learning BlogOfficial· 46/100AI score46/100

Swap-AB kernel and Triton backend speed up MoE inference on Hopper GPUs

Original titleMixture-of-Experts kernel engineering on Amazon SageMaker AI

AISummary

AWS Machine Learning Blog describes a Swap-AB kernel change that flips the MoE matrix multiplication to match Hopper WGMMA layouts.

The change improves end-to-end output throughput by 2–4 percent across large MoE models on H200 and is available in vLLM 0.24.0 or later.

For FP8 tensor-quantized models on Hopper, the post says the Triton backend consistently outperforms the default FLASHINFER_CUTLASS path.

Read the original aws.amazon.com

Source: AWS Machine Learning Blog · aws.amazon.comPublished · added here