From the source
Amazon describes kernel optimizations for fused Mixture-of-Experts models on SageMaker AI, focusing on a Swap-AB technique that transposes matrices to improve WGMMA utilization on
NVIDIA Hopper GPUs during the decode phase.
From the source
From the source
Optimizing MoE kernels on Hopper GPUs via matrix transposition
Amazon describes kernel optimizations for fused Mixture-of-Experts models on SageMaker AI, focusing on a Swap-AB technique that transposes matrices to improve WGMMA utilization on
NVIDIA Hopper GPUs during the decode phase.
From the source
By programmatically flipping the matrix multiplication before it reaches the PTX assembly layer, we realigned the software to match the exact topography of the Hopper silicon.
aws.amazon.com