旧旧阅读网

手机浏览器扫描二维码访问

商汤x记忆张量:国产算力逆袭a100的关键一战(第1页)

最近商汤大装置和记忆张量联手搞出了个大动静——落地了业界首个国产GPGPU PD分离商用推理集群,最亮眼的成绩是:在大模型推理这个关键赛道上,综合推理性价比直接干到了同代NVIDIA A100的150%。

这话乍一听有点抽象,说白了就是:用国产的芯片和软件,跑大模型推理业务,花同样的钱能跑出1.5倍的效果,或者说跑出同样的效果,只需要花三分之二的钱。更关键的是,这不是靠堆硬件、拼参数的“笨办法”,而是靠体系级的软硬件协同创新,给国产算力闯出了一条“弯道超车”的路子,不是跟在别人屁股后面模仿。

下面咱们用大白话把这件事的来龙去脉、核心门道和行业意义掰扯清楚,让大家明白这波操作到底牛在哪。

一、先搞懂:大模型推理为啥这么难?PD分离又是啥?

要理解这个合作的价值,得先明白大模型推理的核心痛点。咱们平时用ChatGPT、文心一言这类大模型,看似只是发个指令等回复,背后的计算过程其实分两大步,这两步的需求天差地别,也是卡住很多算力的“瓶颈”。

第一步叫Prefill(预处理)。当你输入一个问题,比如“帮我写一篇关于古蜀文明的短文”,大模型要先把这段文字转换成机器能看懂的向量,然后一次性计算出大量的中间结果,也就是KV缓存。这个过程的特点是计算量大、可以批量处理,就像工厂里的流水线批量生产零件,需要算力集中发力,但对延迟要求不高——稍微等个几十毫秒,用户根本感觉不到。

第二步叫Decode(解码)。大模型拿到KV缓存后,要逐字逐句生成回复,也就是“逐token生成”。比如先出“古蜀文明”,再出“是中国西南地区一支极具特色的古代文明”,每生成一个词,都要用到前面的KV缓存。这个过程的特点是计算量小,但对延迟要求极高——要是生成一句话卡个一两秒,用户体验直接拉胯,而且这个过程没法批量,只能“串行干活”。

这两步本来是在同一批硬件上完成的,问题就出在这:Prefill阶段需要“暴力计算”,会占满硬件的算力和显存;等轮到Decode阶段,硬件又闲着大半,算力利用率直接暴跌。尤其是国产GPGPU,本身在显存容量、生态适配方面和国际顶尖产品有差距,这么一折腾,劣势更明显——跑同样的任务,成本高、速度慢,根本没法商业化。

那PD分离是啥?说白了就是“分工合作”——把Prefill和Decode这两个阶段,拆到不同的硬件集群上分别处理。负责Prefill的集群专门干“批量重活”,把KV缓存算好;负责Decode的集群专门干“精细快活”,用现成的KV缓存快速生成回复。这样一来,硬件不用在两种模式间来回切换,利用率能提一大截。

但这里要划重点:单纯的硬件PD分离,解决不了根本问题。很多厂商之前也试过,结果发现只是把“一个瓶颈”变成了“两个瓶颈”——Prefill集群算完的KV缓存,要传到Decode集群,传输过程会产生延迟;而且KV缓存本身占显存,就算分开处理,显存不够用的问题还是没解决,性价比提升非常有限。

二、核心突破:不是改硬件,是重构整个推理范式

商汤和记忆张量的聪明之处在于,他们没有局限在“硬件层面拆分工序”,而是把PD分离技术,和记忆张量的核心产品MemOS的激活记忆体系深度绑在了一起,相当于从“流水线分工”升级到了“全产业链协同”,直接重构了大模型推理的底层逻辑。

咱们来拆解这个“协同创新”的关键操作,其实就两件核心事,件件都戳中痛点:

1. MemOS的激活记忆体系:让KV缓存“活起来”,省下大量显存

前面说过,KV缓存是大模型推理的“刚需”,但它特别占显存——一个千亿参数的大模型,一次推理产生的KV缓存,就能吃掉好几GB的显存。传统模式下,这些缓存用过一次就扔,下次再推理,又要重新计算,既费算力又费显存。

而MemOS的激活记忆体系,说白了就是给KV缓存搞了个“智能管理系统”。它能做到两件事:

- 缓存复用:把高频出现的KV缓存(比如用户经常问的“大模型是什么”这类通用问题的缓存)存起来,下次再有用户问类似问题,直接调用现成的缓存,不用重新计算。这就像餐厅里的预制菜,不用每次都从零开始炒菜,节省大量时间和燃气。

- 动态压缩:对不常用的KV缓存,进行无损压缩,把显存占用率降下来。比如原本占10GB的缓存,压缩后只占5GB,省下的显存就能用来跑更多的推理任务,相当于一台机器当成两台用。

热门小说推荐
锦堂春

锦堂春

《锦堂春》作者:蓝家三少简介容九喑第一眼见着那小姑娘的时候,就生出了不该有的念头,娇滴滴的小姑娘,扑到了他腿上,奶声奶气的喊了声,“阿哥!”忽然有一天,小姑娘被他吓哭了,跑得远远的,如风筝断了线。可那又如何?腐朽生花,彼岸黄泉,他都没打算放过她!第1章小叔与寡妇外头敲锣打鼓的,将,军府二公子娶妻,好生热闹。忽然间,新嫁娘的盖头被一...

猫的旅馆APP

猫的旅馆APP

夏安安是位孤独症儿童,四岁半那年刚随妈妈搬去新的住处,她的世界突然多了一款名叫《猫的旅馆》的APP。 这天,一只白猫路过夏安安家门口。 【你的旅馆出现一位潜在客户,请设置舱位,等待客户入住。】 夏安安搬来废弃的纸箱放在后院,一只小橘猫溜进箱子里打滚。 【恭喜你,你的流浪猫旅馆有了第一位住客,入住时长:5分钟,奖励5个金币,请问是否提现?】 夏安安逐渐发现,她真的拥有一个猫的旅馆。 …… 每年冬天,幸福小区周边的流浪猫都会因为寒冷和饥饿折损不少成员,今年自从夏安安和她妈妈搬来以后,它们有吃有住,来年春天,猫群不但全部存活,还多了几窝小猫崽。 猫群决定将夏安安列为重点守护对象,陪伴她成长。 夏安安上幼儿园了,猫群轮班送她上学,蹲在窗外守护,风雨无阻。 金渐层元宝兴奋道:“今天轮到我送那孩子去幼儿园了!她进去之前摸了一把我的脑袋!她摸我哎!” 狮子猫莱茵高傲地“哦”了一声,然后说道:“她昨天搂着我滑滑梯了,感觉还挺不错的。” 橘猫多崽翻了个白眼:“这有啥了不起的,她有本画册,画的可都是我。哦,有一幅是她抱着我的画面。” 众猫:“……” 安安妈妈发现,女儿在猫咪们的陪伴下,孤独症逐渐治愈,还学会主动与人沟通交流,拥有不少好朋友。 阅读指南: 1.温馨治愈日常向动物萌宝文。 2.作者亲测,本文绝对是治疗失眠的绝佳读物,看一次困一次,建议睡前阅读。狗头.jpg 3.关于感情线:孤独症VS社牛,感情线偏后期,18岁后会有感情部分展开。...

TP分叉口

TP分叉口

爱与被爱只在一念之间,无数的分叉路口我们遗忘徘徊,再回首时,和你携手前行的还是当初人生若只如初见的那个人吗?傅琅熙,傅氏集团继承人,22岁留学归来开始掌管傅氏,几年时间让傅氏集团从b市的地产巨...

综影视:过分觊觎

综影视:过分觊觎

短篇小甜饼/穿越影视/主角不定/每个世界的故事都是独立的/并不连贯大家可以当单独的故事看/......

暗域迷踪之风衣人

暗域迷踪之风衣人

人类文明在重建后进入了一个全新的时代,第九观测者等人成为了人类的英雄和守护者。然而,他们并没有停下探索的脚步。他们组建了一支强大的科研团队,深入研究量子科技和宇宙奥秘。他们前往宇宙深处,探索那些未知的星球和文明。在这个过程中,他们遇到了各种各样的挑战和机遇,与其他文明进行交流和合作,共同推动宇宙文明的发展。同时,他......

他笑时风华正茂

他笑时风华正茂

那并不是她第一次遇见他。 只是那次,KTV走廊,他搂着怀里的女孩笑的风情万种。 她印象太深刻。 后来呢? 后来—— 男友陆司北豪请狐朋狗友,他来晚了。 有人喊:“来晚了罚酒啊。” “怎么没带女朋友?” 那人淡淡的一笑,“分了。” “嗬,这次几个月?” 那人抬眼,声音清淡:“够了啊。” 包间里,灯光半明半暗。孟盛楠低着头喝可乐,渗进舌头上却又凉又苦。那时候,她还没意识到,在后来竟会真的非他不可了。...