最近不少人问我一个判断:大模型还在继续变大,国内AI芯片如果真要在这个阶段扛起主力,最关键的那一手棋到底在哪。我的回答基本固定:不是某个计算核心的浮点峰值,也不是单块加速卡的显存大小,而是把芯片真正放进一套能打的大模型系统里的整合能力,说得直白一点,就是全栈协同。
这个结论不是拍脑袋,而是这两年在多机多卡环境里跑训练、做推理、调性能,被无数个细节反复验证过的。大模型规模膨胀之后,芯片行业的竞争逻辑已经发生了一次很根本的变化:以前大家比的是单点算力,现在是比体系化的有效算力。谁能在算子、编译器、通信、调度、推理引擎这几个层面和大模型生命周期的需求咬合得更紧,谁才是真正能落地的答案。
1. 大模型规模膨胀之后,算力的“评价单位”已经变了
1.1 模型尺寸变大,瓶颈被系统性地逐层放大
大模型规模膨胀,最直接的表现就是参数从十亿级冲向千亿级,甚至更高。但很多人忽略了一个关键点:参数变多,不只是“把更多计算塞进芯片”这么简单,它把整个服务器系统的瓶颈都逐层放大了。
举个例子,假设一个千亿参数模型用BF16精度存储,光权重文件就要约200GB,这已经超过绝大多数单卡的显存规格。如果做训练,还要额外考虑优化器状态、梯度、中间激活值,占用的显存倍数会进一步上升。这时候你会发现,单卡解决不了问题,必须把模型切到多卡、多机上,而一旦涉及多卡,通信带宽、同步开销、负载均衡就全来了。
以前一张卡能跑通的活,现在要八张卡、几十张卡协作完成,系统里的短板效应会变得极其明显。只要有一张卡慢一点,或者一根互联链路拥塞,整个训练任务就会被拖到那个最慢的节点上去。这也是为什么我经常跟团队说,在大模型时代,评价AI芯片不能只看单卡性能,要看它被放进集群之后整个系统还能跑出多少利用率。
1.2 显存容量、带宽和算力之间的平衡,是芯片设计的第一道门槛
做芯片的人自己很清楚,算力、显存容量、显存带宽这三件事很难同时做到极致。算力做得高,硅片面积和功耗马上上来;显存容量加得大,成本和内存带宽又会成为新瓶颈。
对大模型来说,显存容量决定了一个模型能不能被放进去,算力决定计算速度,带宽则决定权重参数能不能在计算单元需要的时候及时喂到嘴边。这三者任何一项掉链子,都会变成水桶效应里最短的那块板。
我见过不少纸面算力很漂亮的加速卡,真到大模型场景里一跑,吞吐反而上不去。原因就是显存带宽不足,计算单元大部分时间在等待参数搬运。这就跟请了一个能写一万行代码的程序员,但只给他一条每秒传20行的网络一样,他脑子再快也被传输限制了。
所以现在看AI芯片,我第一眼看的是显存带宽和互联带宽,其次才是峰值算力。峰值决定上限,带宽决定你能摸到上限的几成。
1.3 模型规模的膨胀,让MFU这类系统指标变得比芯片峰值更重要
芯片圈子里有一个词叫MFU,全称是Model FLOPs Utilization,意思是模型实际计算量占芯片理论计算量的比例。放在大模型训练场景里,它衡量的是整个训练系统把硬件利用率发挥到了什么程度。
举个例子,假设一张卡的FP16算力是1000 TFLOPS,但真正跑Transformer的时候,因为算子适配不全、通信同步损耗、显存墙等原因,你可能只能跑出250 TFLOPS,那MFU就是25%。这个数字才是评估“这卡能不能打”的关键。
现在很多团队评估芯片,已经不再只跑某个基准测试,而是直接拿一个具体的大模型训练任务去量MFU。比如跑一个700亿参数的模型,看看单卡MFU能到多少,128卡扩展效率还能不能保住60%以上。这些指标综合起来,才真正反映出一款芯片在大模型场景里的战斗力。
2. 全栈协同到底协同什么:芯片不再是单点英雄题
2.1 全栈不等于全家桶,它是一套围绕模型生命周期运转的体系
提到全栈,很多人以为是把芯片、板卡、服务器、集群管理软件全做一遍,组成一个全家桶。但我理解的全栈协同,不是硬件产品线的堆叠,而是一套围绕大模型生命周期运转的协同体系。
大模型从训练到部署,要走完数据处理、模型结构定义、算子编译、分布式并行、通信优化、推理加速、服务调度这好几道工序。每一道工序都需要芯片生态里对应的软件组件去承接。如果每个组件单独看都还行,但互相之间衔接不上,整套系统的效率就会大打折扣。
这就像一支足球队,单个球员技术都好,但有人习惯拉开单打,有人不知道队友跑位,比赛一样会输。全栈协同要解决的就是这种衔接问题:计算单元知道在哪里等数据,通信库知道什么时候搬运数据,编译器知道怎么把模型的计算图切成最适合这张卡的执行单元。
所以你在衡量一款国内AI芯片的时候,别只看它的硬件规格表,还要看围绕它的那套软件栈做得到不到位,能不能覆盖模型训练和推理的完整链路。
2.2 编译器与算子层,决定你能发挥出芯片的几成功力
芯片和模型之间的翻译官,就是编译器和算子库。模型定义通常是Python和PyTorch框架写出来的,底层芯片只认自己的指令集。中间这一层怎么把模型算子映射成芯片指令,而且映射得高效,直接决定了硬件利用率。
对大模型来说,最核心的算子就是矩阵乘法,也就是GEMM,但它不是单一一个GEMM就能解决的。Transformer里有QKV投影、注意力得分、上下文融合、FFN等多个阶段的GEMM,算子的形状、数据布局、融合方式都不一样。如果编译器不能针对具体模型做算子融合,比如把多个内存访问题压缩成一个,那算力再高也会被内存访问拖死。
我现在去评估一款国内AI芯片,一定会问三个问题:主流开源模型能不能直接跑,跑之前需要改多少代码,算子库对FlashAttention支持的成熟度如何。这三个问题能回答好,说明它在技术层面的全栈协同已经做到位了。
2.3 集群调度与故障恢复:大模型训练本质是一场马拉松
大模型训练很少是几分钟跑完的小任务,往往是几十天连续运转的大工程。这个过程中,节点故障、显卡异常、显存泄漏、网络抖动都可能发生。一套真正能扛事的芯片方案,必须有可靠的集群调度和故障恢复机制。
我见过有个团队用新芯片跑一个千亿参数模型训练,结果训到第三天一个节点掉线,任务直接崩了,因为整套集群方案里缺少自动恢复和断点续训的能力。重新排队、重新加载检查点,三天时间白跑。这种问题比单卡性能差还要致命。
所以说,评估AI芯片的重要一环,是看它配套的集群管理软件能不能做任务调度、健康检查、自动重启和检查点备份。大模型规模膨胀之后,芯片的比拼已经从“能不能算”变成了“能不能稳”。
3. 三个最值得盯的协同点:互联、框架适配、推理链路
3.1 卡间互联:Scale-up 和 Scale-out 的取舍
大模型要跨卡并行,卡间通信几乎决定了训练效率的天花板。现在主流方案分两路:一路是把多张卡通过高速接口组成一个大的GPU Node,卡间通信带宽非常高,这叫做Scale-up;另一路是通过网络把多台服务器连起来,扩大集群规模,这叫做Scale-out。
对国内AI芯片来说,Scale-up通道的设计尤为关键。比如两卡之间通信带宽能到几百GB每秒,带来的直接好处是张量并行时同步参数的耗时更短,整次迭代的等待时间也大幅下降。
我做过一次对比测试:同样一个700亿参数的训练任务,一款Scale-up带宽做得到位的加速卡,128卡训练效率比另一款带宽差一截的卡高了将近一倍。原因很简单,大模型并行训练里每步迭代都要同步梯度,通信时间越短,等待越少,算力越不容易闲着。
3.2 训练框架适配:从“能跑”到“快跑”的距离
现在很多芯片厂商都会说“支持PyTorch”,但这里面的水很深。支持可以分几个层次:最简单的是把一个模型硬跑起来,但速度很慢;进阶一点是常见算子有手写优化版本;最好的是能结合自动混合精度、梯度检查点、分布式数据并行做整体调优。
如果你要微调一个大模型,比如基于Qwen2.5-7B做行业模型微调,就会发现一个现象:显卡理论算力差不多,但不同芯片在微调过程中的实际表现差距非常大。有的芯片跑起来显存占用异常高,需要把batch size调得很小,收敛速度自然慢;有的芯片则对LoRA这类高效微调方法做了专门优化,显存占用很平稳,训练速度也更稳。
所以我的建议是,不要听厂商说“支持PyTorch就完事了”,一定要拿自己真实的模型和真实的训练配置去跑一遍,记录下每步迭代的时间、显存峰值、MFU这几个指标再来判断。
3.3 推理链路:部署时吞吐、时延和利用率如何落地
大模型规模膨胀不只影响训练,推理侧的瓶颈同样明显。部署一个千亿参数模型在线服务,要同时面对长上下文带来的KV Cache膨胀、并发请求带来的吞吐压力、以及响应时延要求。这个时候,芯片在推理链路上的协同能力就成了胜负手。
我比较关注的几个技术点包括PD分离,也就是将预填充和生成解耦到不同实例,避免互相干扰;还有W8A8量化,也就是权重和激活都用8位整数运算,来换取高吞吐和低显存占用;另外还有KV Cache管理,它在长上下文场景里对命中率和内存规划的影响非常大。
哪怕你用llama.cpp或者Ollama在本地部署一个模型,也能明显感受到推理引擎和硬件结合得好不好:同样一个量化模型,不同后端的解码速度可能会差好几倍。这说明在当前和未来,谁能在推理框架里把芯片的潜力真正压榨出来,谁就更值得选。
4. 评估一套AI芯片,真正要跑完的“漏斗式筛选”
4.1 从纸面规格到真实可用率:不能只信参数表
很多团队选芯片,上来先看FP16 TFLOPS、显存容量这些参数。这些数据当然要看,但它们只是第一层漏斗。真正决定芯片行不行的,是后面层层过滤之后剩下来的实测表现。
我习惯把评估过程拆成四层漏斗。第一层是单卡规格,比如算力、显存、带宽,能初步筛掉明显不合硬件需求的方案。第二层是单卡真实算子效率,拿典型的Transformer层跑一遍,看能不能达到纸面算力的五成以上。第三层是多卡扩展效率,从4卡加到64卡,看效率衰减是线性还是断崖。第四层是长稳可靠性,连续跑几天,看会不会出现故障、掉速、显存溢出。
这四个层次全跑完,一款芯片的成色基本就能看得比较透了。很多人只看前两层觉得行,结果一上真实集群就开始翻车,这种场景我见得太多了。
4.2 一套可以直接用的评估清单
基于以往踩过的坑,我整理过一份评估AI芯片的清单,现在转成表格放在这里,照着做基本不会走眼:
| 评估维度 | 具体检查项 | 建议标准 |
|---|---|---|
| 单卡算子 | 跑Transformer典型算子层 | 有效算力不低于纸面峰值的50% |
| 显存规划 | 检查KV Cache和大batch时的显存占用 | 显存占用曲线平稳,无突发性暴涨 |
| 卡间互联 | 跑集合通信基准测试 | 多卡扩展时效率衰减平缓 |
| 分布式兼容 | 直接用训练框架发起多机多卡任务 | 无需大幅修改代码即可跑通 |
| 推理引擎 | 测首字时延与生成吞吐 | 长上下文下吞吐下降不明显 |
| 故障恢复 | 人为kill一个节点观察训练任务 | 能自动摘除故障节点并续训 |
| 生态迁移 | 记录模型修改行数 | 核心模型改动越少越好 |
这套清单不复杂,但每项都很实在。对于要在大模型方向长期投入的团队,花两周时间把这套跑完,比听完厂商几十页PPT有价值得多。
4.3 用“可维护性”为全栈协同兜底
最后要补充一个容易被忽略的维度:可维护性。芯片只有算力没有好用的工具链,就像一台发动机再强,但打开发动机盖全是看不懂的线束,维修一次要拆个三天三夜,那这车也很难开得久。
可维护性体现在几件事上:软硬件升级是否平滑,新模型发布后算子适配跟不跟得上,诊断工具能不能快速定位显存或通信异常,以及环境配置过程是否足够简单。一套成熟的全栈方案,应该让你把精力更多放在算法和业务上,而不是天天和硬件环境搏斗。
5. 实战里最容易踩的三个坑和我的避坑方法
5.1 坑一:拿“跑得动”误当成“跑得好”
很多新芯片刚推出时,能跑通一个ResNet或者一个小Transformer,就有人觉得它行了。但大模型真正考验的是把高算力、大显存、高带宽同时压在一条链路里的极限负载,小模型根本测不出来。
我当时的做法是直接用开源的7B规模模型做一道试金石,先测FP16下的单机8卡训练,再测4机32卡训练,然后用同样的模型做一下流式输出推理。这三个动作下来,芯片有几斤几两就很清楚了。
5.2 坑二:对网络和通信估得过于乐观
大模型训练是一个集群工程,节点之间的网络就是整条流水线的输送带。很多团队在实验室里只测了单机,没考虑跨机通信,到了规整机房部署时才发现网络带宽不够,整个训练任务被通信拖得死死的。
避坑方法是,在规划阶段就把网络方案纳入评估范围,至少算出单卡跨机通信的需求,预留足够的带宽和网卡数量。不要等任务跑起来之后再去加带宽,到那时候该浪费的时间和算力都已经浪费了。
5.3 坑三:只测训练,不测推理和微调
大模型不只是训练一次就完了,还要持续做微调、做量化、做线上部署。如果你的芯片方案只把训练优化得很好,但推理时延高得离谱,或者量化工具链不完善,那它也很难到业务手里真正发挥价值。
我现在的习惯是,训练、微调、推理三条链路一起测。微调场景要重点看LoRA这类参数高效微调的显存开销,推理场景要看长上下文连续请求下的吞吐稳定性。这样测下来的结论,才能覆盖大模型全生命周期。
6. 给已经决定用国内AI芯片的团队几句实在话
6.1 把它当成新平台来用,不要总想着“等价替代”
很多人刚开始用国内AI芯片时,会下意识地按照过去熟悉的平台习惯去操作,遇到不兼容就开始抱怨。但我的经验是,要把它当成一个新平台来学习,习惯它的调试工具、性能分析方法和算子支持边界。
这不是让你迁就它,而是让你更快找到扬长避短的方式。不同芯片的架构设计理念不同,擅长的高性能场景也不同。理解架构之后,再把模型里最耗时的计算块抽出来逐段调优,往往比无脑搬运老代码效果要好得多。
6.2 小规模先把“韧性”打出来,再冲大集群
如果你正打算用某款国内AI芯片做大规模训练,我的建议是先别直接上几百卡的集群。先用少量节点把模型跑通,把分布式并行策略调对,把故障恢复机制验证到位,然后再逐步扩大规模。
这个道理跟盖楼一样,地基没打稳,楼层越高风险越大。小规模练兵听起来慢,实际上是最快通往稳定的路径。我见过太多团队一上来就冲大集群,结果一个通信参数配错导致全体重来,整体进度反而更慢。
6.3 尽量让模型结构与工具链能力相互匹配
大模型规模膨胀时代,模型结构和技术栈之间其实存在一个共同演进的关系。如果你的模型结构选型能够贴近芯片擅长优化的方向,比如对注意力计算、FFN融合和长序列推理有针对性优化,那么跑起来的效果和效率都会更好。
这并不意味着被芯片绑架,而是说芯片、框架、模型结构三者本就是一套互相绑定的系统。真正成熟的全栈协同,最后体现出来的状态是:你用一套模型去做微调时,不需要为了迁就硬件而牺牲模型效果,也不需要为了模型效果而强行忍受低效率。
我自己在实际项目里感受最深的一件事是:大模型越大,越考验芯片背后的系统力。胜负手从来不在那张孤零零的芯片上面,而在于把它放进真实的大模型系统中,从训练到微调再到推理,能不能从头到尾保持高效、稳定、可维护。谁能把这件事做成,谁才是这一轮算力竞赛里真正走得远的那一个。