CUDA FlashAttention:从在线 Softmax 到长序列 IO 极限
以大模型因果 Attention 为背景,从二次中间矩阵、在线 Softmax、分层 tile 与 warp/MMA 数据流出发,实现并验证 FlashAttention 的 CUDA 优化。












通过 RSS 订阅,第一时间获取最新文章和动态
复制链接到你的 RSS 阅读器
https://chengzipai.xyz/rss.xml以大模型因果 Attention 为背景,从二次中间矩阵、在线 Softmax、分层 tile 与 warp/MMA 数据流出发,实现并验证 FlashAttention 的 CUDA 优化。
面向大模型推理拆解 KV Cache 容量、分页寻址、向量化 append、GQA decode 与 split-KV,并用 FlashInfer 和 PyTorch 做长上下文实测。
把 LSMR 小向量递推融合为 CUDA kernel,并用两组真实 FP32 重建输入对齐 CuPy 的迭代、残差与性能。
拆解 LSMR 中 A·v 与 Aᵀ·u 的稀疏数据通路,解释 CSR/CSC 双存储、行长负载不均和显存池的收益与代价。
从 Golub-Kahan 双对角化出发,拆解一个纯 CUDA FP32 LSMR 库的性能上界、接口边界与分层架构。
从标量基线到 Tensor Core、异步流水、低比特缩放和分段调度:构建一个面向大模型全连接层的工业级 GEMM 算子族。
RSS(Really Simple Syndication)是一种用于发布经常更新内容的标准格式。通过 RSS,你可以:
推荐使用 Feedly、Inoreader 或其他 RSS 阅读器来订阅本站。
暂未播放