Skip to main content
grasp.study
About
Explore
Pricing
Log in
Try it for free
Back to course
👋
Welcome to a Grasp lesson
Create a free account
to keep studying and create your own study path.
High-Performance AI Computing
·
Module 11
Advanced Kernels with Tensor Cores
1
Tensor Cores and `mma.sync` for Matrix Operations
Describe the architecture of NVIDIA Tensor Cores and the `mma.sync` instruction for matrix multiply-accumulate operations.
2
Warp-Level Matrix Multiplication with `nvcuda::wmma`
Use the `nvcuda::wmma` C++ API to load matrices into fragments, perform warp-level MMA, and store results.
3
GEMM with WMMA: From Global to Registers
Construct a complete GEMM kernel using `wmma` intrinsics, managing the data pipeline from global memory to shared memory and finally to registers.
4
Understanding FlashAttention: Memory Complexity
Explain the quadratic memory complexity of standard self-attention that FlashAttention addresses.
5
FlashAttention: Tiling and Online Softmax
Describe the core concepts of FlashAttention: tiling and online softmax computation.
6
FlashAttention Inner Loop Implementation
Implement the inner loop of a simplified FlashAttention kernel, computing a block of the output matrix.
7
FlashAttention Forward Pass Implementation and Verification
Combine components into a simplified FlashAttention forward pass kernel and verify its output against a naive baseline.
Previous module
Core AI Kernels: Fusion, Normalization, and Backpropagation
Next module
Integrating Kernels in the AI Ecosystem