Courses

Interactive teaching materials by Han Tian.

Software Optimization for Intelligent Computing

Chapter 5 · GPU Architecture and Computational Optimization

Select actual tiles within the full Q, Kᵀ, and V matrices. Follow their elements from HBM to on-chip GPU storage, through local computation and online Softmax, and back to the matching output tiles in O.

  • Standard Attention baseline
  • Online Softmax
  • FA1 / FA2 work partitioning