Skip to content
Featured projects TL;DR In this blog post, we present various novel kernel fusion techniques for common normalization ops like LayerNorm and RMSNorm, which provide significant speedup by reducing the memory-IO overhead of these highly memory-bound kernels. We start with a brief overview of the modeling importance as well as performance challenges of normalization ops common in both LLMs and ads re...
Towards Free Normalization: Fusing Normalization into GEMM and Attention Kernels | Huntaegis