专栏名称: 数字生命卡兹克
反复横跳于不同的AI领域,努力分享一些很酷的AI干货
目录
相关文章推荐
51好读  ›  专栏  ›  数字生命卡兹克

DeepSeek开源第三天,只用300行代码就超越了英伟达自己。

数字生命卡兹克  · 公众号  · AI 科技自媒体  · 2025-02-26 10:49

主要观点总结

文章介绍了DeepSeek团队最近开源的DeepGEMM库,该库专门为DeepSeek-V3设计,用于提高AI性能效率。文章提到了DeepGEMM的几个关键点,包括其支持FP8的通用矩阵乘法(GEMM),Mix-of-Experts(MoE)分组GEMM,以及其对CUDA核心的使用来提高计算精度。此外,DeepGEMM具有轻量级即时编译(JIT)模块,无需编译即可在运行时编译所有内核。文章还提到了DeepGEMM在某些情况下的优化表现及改进建议,强调了开源的重要性,并呼吁读者参与改进。最后,文章鼓励读者点赞、在看、转发并给出星标支持。

关键观点总结

关键观点1: DeepGEMM库是DeepSeek团队为DeepSeek-V3专门设计的开源库,旨在提高AI性能效率。

该库支持FP8的通用矩阵乘法(GEMM)和Mix-of-Experts(MoE)分组GEMM。

关键观点2: DeepGEMM利用CUDA核心进行两次累加,以提高FP8这种速度快但精度偏低的计算方式的准确性。

通过即时编译(JIT)模块,无需编译即可在运行时编译所有内核,提高了效率。

关键观点3: DeepGEMM在某些情况下表现优秀,但也存在改进空间。

DeepSeek团队鼓励所有人一起改进这个库。

关键观点4: DeepGEMM的开源理念被强调。

文章提到开源是推动AI发展的重要力量,并鼓励读者参与开源社区,支持开源项目。


正文

请到「今天看啥」查看全文


不能再肝了,但我又觉得DeepSeek值得。。。
这两天,DeepSeek的高强度开源波,一山更比一山高。
先是给GPU安超频加速外挂的 FlashMLA ,又是叫英伟达知道“原来GPU没有商业护城河”的 DeepEP
我也都第一时间给大家带来了报道。
在追求效率、把硬件资源干下来的路上,DeepSeek快成AI性能效率上的Godfather了。。。
这回,他们开源的是一个叫做 DeepGEMM 的玩意儿,专门给当时爆cei全网的DeepSeek-V3做的。
Github星星没半小时,就几百个了。点的越多,意味着开源友友们越喜爱和越关注这个代码仓库,水分那是相当的少 。一般几千的星星就已经算是爆款了,半小时就几百,这个含金量你懂的。
开源链接在此:https://github.com/deepseek-ai/DeepGEMM?tab=readme-ov-file
这东西,倒也没那么难懂。
举个例子,假如我结婚了。场面特别特别大,记得是假如。。。
几百万人组成的迎亲队伍、点鞭炮得点几亿种、接亲队伍也叫个几百万人来,甚至我再搞点大的出来,比如弄个几万盏灯光秀。
所有的一切的一切都需要计算好时间点,相互之间得互相搭配。而DeepGEMM这东西,能把以上所有东西塞进一个矩阵里。
所有迎亲队伍的实时行走轨迹、啥时候点鞭炮的精细时间规划、接亲队伍得到哪里等、等多久,几万盏灯光秀和几千万首音乐秀,几分几秒,该怎么配合,效果最好,等等。
全都能放进矩阵里计算,这都快成在天上俯瞰人间的God了。。。
用技术语言说,就是:
DeepGEMM 是一个为 DeepSeek-V3 专门设计的,用于 FP8 的,通用矩阵乘法(GEMM)库。还支持 普通的和专家混合(Mix-of-Experts,MoE)分组 GEMM
安装时,你都无需编译,只通过一个轻量级的即时编译(JIT)模块,在运行时就可以编译所有内核了。牛逼,一点多余东西都不舍得让你多干活。
而且,只用了300行代码,实在是牛逼。。。
目前,DeepGEMM跟前两天一样,还是只支持H卡。它为了让FP8这种速度快但精度偏低的计算方式变得更准确,利用了CUDA核心做了两次累加。
简单说就是先用FP8完成快速计算,然后再用CUDA核心对结果进行更精细的再加工,这样既能保持速度快,还能把精度提上去。
DeepGEMM也借鉴了英伟达CUTLASS和CuTe的一些概念。
CUTLASS 是基于英伟达明星当家CUDA架构。简单说,它是一个写给 NVIDIA显卡的工具包,专门用来加速“矩阵计算”的。
英伟达的CUTLASS实在是过于高效,以至于被用来构建内核时,几乎能帮显卡把矩阵计算的性能榨到极限,跑到显卡的理论峰值。
但是如果你手里的硬件没那么强大,就像很多现在的AI公司们还停在上一代的卡上时,CUTLASS这种大而全的加速套件,就有点用不上了。
CUTLASS虽然时哥通用、功能强大的矩阵加速库,但是DeepGEMM这种激进的优化方式更专注、更轻量。
深刻的展现了DeepSeek那种“抠”到极致的理念。
把性能也抠到了极限。
性能只要卡的不死,DeepSeek就能拿效率调优这条至简大路冲出来,无形中连美国算力封锁都给捅破了。。。
它完全没有一点对英伟达项目的模版or代数的过分的依赖度,全凭自主。
而且不止是轻量化,性能也是直接起飞。
按他们的话说,
团队说,能够匹配甚至超越英伟达、ADM等等专家专门调优的库。。。
比英伟达自己的 CUTLASS 3.6,速度还提升了2.7倍。
他们在H800上,测试了 DeepSeek-V3 和 R1 推理中可能用到的所有矩阵情况,性能水平,我都整理在这了。
先是密集模型档,估计老黄那个项目的人,也很难想明白,几百行代码怎么调优调成这样的。。。
之前不是都说,硬件是有护城河的嘛。。。现在看起来,DeepSeek比英伟达都懂GPU。
然后就是现在被称为AI未来方向之一的专家混合模型MoE了。它在处理复杂任务上独树一帜。整体的性能,实在是太硬核了。数据如下:
但DeepSeek的人也确实说了。
DeepGEMM虽然非常牛逼,但是在某些情况上的表现确实不太好,欢迎所有人一起改进。
具体的部署上,依旧和之前每次DeepSeek开源时的动作一样,把饭喂到你嘴边,顺便走的时候,再给你擦擦嘴。
因为无需编译,部署速度会更快、更顺畅。这让我想起来了当年的贴吧大神们,只留下宝典教程里最核心的部分,挥挥衣袖,就跑了。。。
随着下一代基座模型,比如DeepSeek V4、GPT-4.5等等的参数和复杂度继续增长时,深入到底层进行优化的DeepGEMM这种库,真的会越来越重要。
AI圈子内曾经充斥着,闭源才是通向AGI的论调。
这平等地伤害了,每一个踏进AI大门的普通人们。
闭源的AI世界,就像是黑暗森林。
每个人都是拿着枪追着篝火的猎人。
但DeepSeek这一举。
让我突然想起来《教父》里那句名言。
永远不要动怒。
绝不要威胁。
要讲道理。
开源就是DeepSeek这群家伙们的道理。
共勉。

以上,既然看到这里了,如果觉得不错,随手点个赞、在看、转发三连吧,如果想第一时间收到推送,也可以给我个星标⭐~谢谢你看我的文章,我们,下次再见。

>/ 作者:卡兹克、芝兰山

>/ 投稿或爆料,请联系邮箱:[email protected]







请到「今天看啥」查看全文