DeepSeek 开源周第一天： FlashMLA

112 天前
 fu82581983

为 Hopper GPU 设计的高效 MLA 解码内核，针对可变长度序列进行了优化，并且投入了生产环境。

✅ 支持 BF16 精度

✅ 使用分页 KV 缓存（块大小为 64 ）

⚡ 在 H800 GPU 上实现 3000 GB/s 的内存带宽和 580 TFLOPS 的计算性能

Github: https://github.com/deepseek-ai/FlashMLA

X: https://x.com/deepseek_ai/status/1893836827574030466

832 次点击
所在节点    分享发现
0 条回复

这是一个专为移动设备优化的页面（即为了让你能够在 Google 搜索结果里秒开这个页面），如果你希望参与 V2EX 社区的讨论，你可以继续到 V2EX 上打开本讨论主题的完整版本。

https://ex.noerr.eu.org/t/1113750

V2EX 是创意工作者们的社区，是一个分享自己正在做的有趣事物、交流想法，可以遇见新朋友甚至新机会的地方。

V2EX is a community of developers, designers and creative people.

© 2021 V2EX