Designing High-Performance GPU Kernels with TileLang: Tensor-Core GEMM, Fused Softmax, FlashAttention, and Autotuning
In this tutorial, we discover TileLang as a high-level Python domain-specific language for designing and compiling performance-oriented GPU kernels by TVM. We start by validating the CUDA atmosphere and establishing reusable benchmarking and numerical-verification utilities, then progressively implement vector addition, tiled tensor-core matrix multiplication, schedule exploration, fused GEMM epilogues, row-wise softmax, and FlashAttention. Throughout the…
