GitHub · 去对齐工具与仓库·· 4 小时前AI 评分52
miniflash-triton:Triton 实现的高性能 LLM 注意力与推理核开源
sarimahsan/miniflash-triton
来源与原文
GitHub · 去对齐工具与仓库
发布时间:(北京时间)
阅读原始来源 · github.com ↗完整链接与内容说明
https://github.com/sarimahsan/miniflash-triton
本站目前仅有 AI 导读与摘要,未展示原文正文;技术细节、代码和完整结论请查看原始来源。
AI 导读
sarimahsan 开源 miniflash-triton,用 Triton 实现 FlashAttention-2、Split-KV FlashDecode、零拷贝 GQA、分页 KV-cache 与 CUDA Graph 推理核。在 A100、L4 和 RTX PRO 6000 Blackwell 上对比 PyTorch SDPA:Blackwell 前向注意力达 264.1 TFLOP/s,S=1024 时延迟降低 1.41x;A100 解码带宽 1,452 GB/s(达 HBM 物理上限的 93.4%);GPT-2 端到端解码 961.5 tok/s(较 eager SDPA 提升 9.1x),Llama-1B 达 322.5 tok/s。
来源:GitHub · 去对齐工具与仓库 · github.com