44 节免费 GPU 内核工程系列:从朴素 GEMM 讲到 FlashAttention 覆盖 H100/Blackwell

Winter_Mistake_3185 · reddit · 2026-10-20

一位开发者在 Reddit 发布了免费、代码优先的 GPU 内核工程系列课程(计划 44 节),已放出导论与路线图视频,全部代码公开,明确不以售卖为目的。

核心理念:对多数 ML 负载,内核性能首先是数据搬运问题,其次才是计算问题。全系列围绕两条主线展开:

路线图涵盖:CPU 并行与 GPU 架构基础、内存层级、GEMM 从朴素到优化(每步配合 profiling)、Tensor Cores、profiling 与调试、PyTorch 集成与 Triton、FlashAttention,以及 Hopper/Blackwell 专题(TMA、WGMMA、CUTLASS、FP8/量化、NVFP4、多卡基础、FlashAttention 4、DeepSeek 内核、AI 写内核与毕业项目)。

每节课都是「概念 → 现场写码 → profile → 对照 baseline 验证确实变快」。前置要求为 C/C++、基础线性代数和 PyTorch。作者注明后期课程(尤其 Blackwell/NVFP4)可能随工具链演进调整,并邀请有生产内核经验的开发者反馈课程遗漏点。

原文链接 →

「Infra」频道最新

更多「Infra」频道 AI 资讯 →