One Trip, No Detours
A naive PyTorch implementation runs elementwise add, then a bias add, then a GELU activation as three separate CUDA kernel launches, each reading and writing the full tensor to HBM. What does fusing these into a single custom kernel primarily improve, and why?
Sign in to answer questions and track your progress
Sign In