RMSNorm未必比LayerNorm快多少
cHHillee · x · 2026-07-13
作者认为，关于 RMSNorm 比 LayerNorm “便宜很多”的说法被过度简化了。 核心观点是：这类算子在很多场景下都是 **memory-bound kernel**，真正耗时往往在把数据搬到 GPU 核心，而不是算术本身，所以二者端到端耗时可能接近。原论文对这类内存瓶颈的理解并不充分，因此结论有偏差。 但作者也补充说，在把性能继续往极限推、例如做 matmul epilogue 融合时，LayerNorm 因为要额外跟踪 mean shift，确实会更难融合，因此也不能简单说它“完全免费”。结论是：RMSNorm 不一定比 LayerNorm 快两倍，但 LayerNorm 也不是零成本。