LogoCookLLM文档
LogoCookLLM文档
首页CookLLM

原理精讲

词元化
Tokenization 基础BPE 算法详解GPT 系列 TokenizerBPE 训练工程化
模型架构
Attention 机制详解
Engram
GPU 编程基础
GPU 架构基础张量布局Triton 入门:向量加法
FlashAttention
Flash Attention 原理详解从朴素实现到 Auto-TuningBlock Pointer 与多维支持Causal Masking 优化Grouped Query Attention反向传播实现

动手训练

系统工程FlashAttention

Block Pointer 与多维支持

会员专享

从单序列扩展到 Batch/Head 并行,并使用 Block Pointer 简化指针管理。

配套代码

登录以继续阅读

这是一篇付费内容,请登录您的账户以访问完整内容。

从朴素实现到 Auto-Tuning

编写第一个 Flash Attention Kernel,并利用 Auto-Tune 进行性能优化。

Causal Masking 优化

为自回归模型实现因果注意力机制,通过跳过上三角计算实现 ~2x 加速。

目录

从单序列到 Batch/Head 并行
4D 张量的内存布局
3D Grid 并行化
手动指针偏移
Block Pointer:优雅的解决方案
核心 API
循环中的指针移动
完整对比
总结