跳到主要内容

技术

「技术」类别的杂记 · 共 4 篇

FlashAttention:把 O(n²) 显存压成 O(n)

不换算法,只靠分块与重计算,让标准注意力在长序列上快一个量级。

Markdown 语法演示

本站支持的 Markdown 语法效果一览:文字样式、列表、引用、代码、表格、公式与 Mermaid 图。

Triton 入门:写第一个 kernel

用 Triton 重写向量加法,理解 program、tl.arange 与 block 启动背后的模型。

GPU 性能分析工具地图

nvidia-smi、PyTorch Profiler、Nsight Compute 各自看什么,先看哪个指标。