丛编
杂记
更多
「技术」类别的杂记 · 共 4 篇
不换算法,只靠分块与重计算,让标准注意力在长序列上快一个量级。
本站支持的 Markdown 语法效果一览:文字样式、列表、引用、代码、表格、公式与 Mermaid 图。
用 Triton 重写向量加法,理解 program、tl.arange 与 block 启动背后的模型。
nvidia-smi、PyTorch Profiler、Nsight Compute 各自看什么,先看哪个指标。