• 元宇宙:本站分享元宇宙相关资讯,资讯仅代表作者观点与平台立场无关,仅供参考.

“源神”DeepSeek 突破H800性能上限 FlashMLA重磅开源 算力成本还能降

  • 2025年2月24日 20:13

来源:量子位

DeepSeek开源周第一天,降本大法公开——

FlashMLA,直接突破H800计算上限。

它是为HopperGPU开发的高效MLA解码内核,专门针对可变长度序列进行了优化,目前已经投入生产。

MLA,正是DeepSeek提出的创新注意力架构。从V2开始,MLA使得DeepSeek在系列模型中实现成本大幅降低,但是计算、推理性能仍能与顶尖模型持平。

按照官方介绍来说,FlashMLA使用之后,H800可以达到3000GB/s内存,实现580TFLOPS计算性能。

网友们纷纷点赞:向工程团队致以崇高的敬意,从Hopper的张量核中挤出了每一个FLOP。这就是我们将LLM服务推向新前沿的方式!

已经有网友用上了。

开源第一天:FlashMLA

目前GitHub页面已经更新。短短一小时,Star星数已经超过1.2k。

此次已经发布:

支持BF16;

分页KV缓存,块大小为64

快速启动:

FlashAttention是能实现快速且内存高效的精确注意力,主流大模型都有在用。最新的第三代,可以让H100利用率飙升至75%。训练速度提升1.5-2倍,FP16下计算吞吐量高达740TFLOPs/s,达理论最大吞吐量75%,更充分利用计算资源,此前只能做到35%。

核心作者是TriDao,普林斯顿大牛,TogetherAI的首席科学家。

而英伟达CUTLASS是CUDAC++模板抽象的集合,用于在CUDA内实现高性能矩阵-矩阵乘法(GEMM)和所有级别和规模的相关计算。MLA,DeepSeek基本架构

最后再来说说,MLA,多头潜在注意力机制,DeepSeek系列模型的基本架构,旨在优化Transformer模型的推理效率与内存使用,同时保持模型性能。

它通过低秩联合压缩技术,将多头注意力中的键(Key)和值(Value)矩阵投影到低维潜在空间,从而显著减少键值缓存(KVCache)的存储需求。这种方法在长序列处理中尤为重要,因为传统方法需要存储完整的KV矩阵,而MLA通过压缩仅保留关键信息。

V2版本中,这一创新性架构把显存占用降到了过去最常用的MHA架构的5%-13%,实现了成本大幅降低。它的推理成本仅为Llama370B的1/7、GPT-4Turbo的1/70。

而在V3,这一降本提速就更为明显,直接让DeepSeek吸引全球目光。

Thewhaleismakingwaves!鲸鱼正在掀起波浪!

好了期待一下,接下来的四天会发些什么呢?

Copyright © 2021.Company 元宇宙YITB.COM All rights reserved.元宇宙YITB.COM