云端 AI 芯片
3 秒看懂
云端 AI 芯片是部署在数据中心、专门用于人工智能训练与推理的大规模并行处理器。与终端芯片不同,它追求极致的算力密度、内存带宽和横向扩展能力,典型形态包括 GPU、TPU/NPU 等 AI 专用加速器。当前技术制高点聚焦于 Chiplet/先进封装、HBM 高带宽内存、以及支持千亿/万亿参数大模型的互联体系。算力军备竞赛由英伟达主导,但云厂自研 ASIC 与新兴架构正引发格局重塑。
3 分钟产业解释
云端 AI 芯片的本质,是把“训练一个 ChatGPT 式模型”或“每秒处理上亿次推理请求”所需的计算、访存、通信三大能力,浓缩在一个或一组可伸缩的硬件单元里。训练阶段要完成对一个巨型神经网络的海量矩阵乘法与梯度同步,推理阶段则要在低延迟、高并发下运行一个大模型的前向传播。两者都对芯片提出严峻挑战:
- 算力:需要大量的乘加单元(MAC)并行化,数据格式从 FP32 向 TF32/FP16/BF16/INT8/FP8 持续下探,以提升能效比。
- 访存:参数与中间激活远超单颗芯片 SRAM 容量,必须依赖外部高带宽内存(如 HBM),其带宽往往达到 TB/s 级别,是卡算力的第一瓶颈。
- 通信:单一芯片算力有限,必须通过高速互联(NVLink、PCIe、InfiniBand、专用 ICI)把数千至数万颗芯片连接成一个整体,执行 AllReduce、All-to-All 等集合通信。
产业形态上,云端 AI 芯片不再只是“高算力 GPU”,而是一个集芯片、互连、系统、生态于一体的完整平台。英伟达通过 GPU+CUDA+NVLink+InfiniBand 构筑了极深的护城河。谷歌 TPU 证明了数据中心内部从训练到推理全用 ASIC 的可行性,并催生了亚马逊 Trainium/Inferentia、微软 Maia、Meta MTIA 等云厂自研浪潮。另一边,初创公司(如 Cerebras、Groq、Graphcore)试图用晶圆级芯片、确定性路由等极限创新打破常规,但商业规模化仍是考验。
15 分钟专家深入
在“大模型 scaling law”驱动下,云端 AI 芯片已从单纯的单芯片性能竞赛,演变为“芯片-封装-节点-集群”全栈协同的系统工程。以下几个维度是理解当前产业博弈的关键:
-
算力标称与实际利用率
FP16/BF16 标称算力(单位 TFLOPS/PFLOPS)只是上限。模型的矩阵乘是否能填满计算单元,取决于 batch size、序列长度与算子融合程度。实际可达算力(Model FLOPS Utilization, MFU)在超大模型中通常仅 30%-60%。FlashAttention、混合精度、持续批处理等技术正是为了逼近算力峰值。 -
内存墙与 HBM 进阶之路
单个推理请求可能仅需几 MFLOPS,但模型权重全部驻留在内存,推理的吞吐量完全由内存带宽决定。训练中,优化器状态、梯度、参数副本使得内存需求是参数量的数倍。HBM 通过 3D 堆叠与硅中介层(Interposer)大幅提升带宽与容量,但成本高昂,且被少数厂商垄断。HBM 的代际迭代(容量/带宽增长)与先进封装(CoWoS 等)的产能,直接决定高端 AI 芯片的供给。 -
互联拓扑与通信代价
超级训练集群需要多维度并行:数据并行、张量并行、流水线并行、序列并行,再加上 MoE 的专家并行。张量并行在每个 Transformer 层内切开矩阵乘,需要在层内进行 AllReduce/ReduceScatter,对互联带宽和时延要求最高,通常局限于高带宽域(如 NVLink 域)。而 MoE 的 All-to-All 分发/汇聚在跨节点通信上带来巨大压力。新一代芯片普遍将高速 SerDes 与专用协议集成进 chiplet 或主芯片,试图将跨机架通信“近程化”。 -
从硅到液的能效工程
单颗芯片 TDP 已从 400W 向 1000W+ 迈进,单机架功耗可破 100kW,传统的风冷被迫让位于液冷(冷板式或浸没式)。芯片设计内部必须支持动态电压频率调节、门控时钟、算存休眠等细粒度电源管理,甚至采用垂直供电架构(背面供电)。能效(TOPS/W)不仅是运营成本,更是数据中心受电容量限制下的准入证。
技术原理
云端 AI 芯片的底层可以抽象为一个“张量计算引擎 + 可编程数据流”。以训练一个 Transformer 大模型为例,核心计算原语包括:
- 通用矩阵乘(GEMM):占据 90% 以上运算时间。芯片内布满由乘加单元组成的脉动阵列或分布式计算簇,从本地缓冲区/共享内存中读取数据,按固定节拍完成大规模并行乘加。
- 非线性激活与归一化:点运算(如 GELU、LayerNorm)需要矢量单元处理,计算密度低但访存密集,往往通过算子融合与矩阵乘共用一个数据流,减少显存访问。
- 注意力计算:Softmax 前的大矩阵乘及后续的mask + Softmax + dropout,涉及多轮读写。FlashAttention 通过分块 (tiling) 和重计算策略,将中间注意力矩阵保持在 SRAM 内,大幅减少 HBM 访问。
- 集合通信:反向传播后,需要将各芯片上的梯度求和(AllReduce)。典型实现基于 Ring 或 Tree 算法,芯片间通过高速链路段到段传递,利用带宽最优的 ReduceScatter+AllGather 组合,通信量压缩至
2(N-1)/N倍梯度体积。
以 N 个 GPU 的数据并行为例,梯度同步阶段所需的时间约等于:
T_{\text{comm}} \approx \frac{\text{Gradient Size}}{\text{Bandwidth}} \times f(N)
f(N) 与拓扑有关,对于最简化的全双工环,趋近于 2(N-1)/N。因此在千卡级以上,网络带宽与架构几乎必须与计算等比升级。
典型芯片微架构(定性示意):
┌────────────────────────────────────────────────────┐
│ AI 芯片 (die) │
│ ┌──────────┐ ┌──────────┐ ┌──────────┐ │
│ │计算核心A │ │计算核心B │ │计算核心C │ ... │
│ │ (MAC阵列) │ │ (MAC阵列) │ │ (MAC阵列) │ │
│ └───┬──────┘ └───┬──────┘ └───┬──────┘ │
│ │ 片上网络 (NoC) / 交叉开关 │ │
│ ┌───┴─────────────────────────────┴──────┐ │
│ │ 全局共享 SRAM / L2 Cache │ │
│ └───────────────────┬────────────────────┘ │
│ Memory Controller (HBM PHY) │
└─────────────────────┬──────────────────────────────┘
┌───┴───┐ HBM Stack
│ HBM │ (多层DRAM堆叠)
└───────┘
Chip-to-Chip I/O (SerDes/ NVLink/ UALink 等)
技术演进史
-
2012-2015:GPGPU 一枝独秀
深度学习崛起初期,NVIDIA GPU 凭借可编程性和 CUDA 生态迅速成为事实标准。云端训练从 K80 到 P100,逐步加入 FP16 精度的硬件支持,但仍是通用 GPU 改良。 -
2016-2018:AI 专用架构破冰
谷歌 TPU v1(仅推理)公布,脉动阵列专攻矩阵乘,能效远超同期 GPU。TPU v2/v3 加入训练能力,并引入 TPU Pod 超级计算机概念。同期,启动 Graphcore IPU、寒武纪思元等创业潮,强调片上存储和不同的并行模型。 -
2019-2022:大规模扩展与互联升级
NVIDIA V100/A100 引入 Tensor Core 和 NVLink/NVSwitch 拓扑,容量与带宽翻番。HBM2e 普及,CoWoS 封装成为高端标配。云厂模仿 TPU 路径,亚马逊推出 Inferentia、Trainium,百度昆仑、阿里含光等现身。训练规模从数百卡膨胀到数千卡,分布式训练框架(Megatron, DeepSpeed)成熟。 -
2023-至今:万亿参数时代的平台战争
NVIDIA H100/H200/B100/B200 系列通过 Transformer Engine、FP8、更大 HBM、更强 NVLink Network 将集群算力推向 ExaFLOPS 级。GPT-4 类大模型爆发,引爆 MoE 架构,带来 All-to-All 通信的新复杂度。云厂加速自研 ASIC,谷歌 TPU v5p 大规模部署,微软 Maia、Meta MTIA 跟进。同时,后端网络从 InfiniBand 转向融合以太网 (RoCEv2/Ultra Ethernet),UALink、Ultra Accelerator Link 等联盟试图打破 NVLink 封闭生态。Cerebras 用晶圆级芯片实现单颗芯片训练一个 GPT-3 大小的模型,Groq 推确定性的流式推理。技术演进进入系统级、生态级对抗。
技术路线对比
| 维度 | GPU (英伟达主导) | 自研 ASIC (TPU/Trainium等) | 晶圆级芯片 (Cerebras) | 其他新型架构 (Groq等) |
|---|---|---|---|---|
| 计算能力 | Tensor Core 通用性高,标称峰值强劲,FP8/FP4 快速迭代 | 针对自家负载极致调优,峰值可能保守但利用率高 | 晶圆集成超大规模核心,免跨芯片通信,标称算力极大 | 各有特色:流式存内计算、确定性调度等 |
| 内存与互联 | 率先采用最新 HBM 代际,NVLink/NVSwitch 构建高带宽域,InfiniBand/Spectrum-X 组集群 | HBM/封装跟进主流,互联多基于以太网或定制 ICI,跨机通信依赖云基础设施 | 片上 SRAM 巨大,无传统 HBM,内存容量受限,跨系统通信特殊设计 | 各不相同,部分依赖专用内存或 SRAM 重构 |
| 软件生态 | CUDA 生态最强,开发者黏性极高,生态壁垒深厚 | 自研编译器与框架,深度绑定内部业务,开放性不足 | 专用 SDK,需重构软件栈,适配成本高 | 需要从编译器底层重建编程模型,生态极弱 |
| 灵活性 | 支持几乎全部模型架构,可适配多变的研究需求 | 可固化特定算子,但架构演进速度快需灵活调整 | 支持大模型但受限于内存大小,适用于特定尺寸 | 通常针对推理或特定训练范式 |
| 成本结构 | 单位算力定价最高,但开发效率带来的总拥有成本(TCO)有竞争力 | 不计研发分摊的情况下,大规模部署的硬件成本显著低于同等 GPU | 极贵,但节省通信成本,系统总体简化 | 量产后成本有望极低,但前期投入巨大 |
| 供应稳定性 | 受制于 HBM、先进封装产能,市场紧缺 | 产能同样受限于代工厂先进制程与封装,但由于非市场主力竞品,获取产能难度不一 | 依赖自己的定制工艺与封装,限制缩放 | 许多采用特殊工艺,供应链脆弱 |
上下游
上游
- EDA/IP:芯片设计工具与高性能接口 IP(HBM PHY、高速 SerDes、PCIe、CXL)必不可少,主要由 Cadence、Synopsys、Alphawave 等提供。
- 代工与制程:5nm/4nm/3nm 逻辑工艺是当前主流,台积电、三星、英特尔提供。先进制程朝向 GAA(Gate-All-Around)晶体管演进,以提升能效。
- 先进封装与基板:CoWoS、EMIB、Foveros 等 2.5D/3D 封装技术,配套 ABF 载板、硅中介层,直接决定 HBM 集成能力和产能。
- HBM 内存:三星、SK 海力士、美光垄断,从 HBM2e 到 HBM3/HBM3e 持续演进,每一代提升带宽与堆叠层数。
- 测试与设备:AI 芯片 die 面积大、功耗高,对 ATE 测试、热验证、系统级测试提出极高要求。
下游
- 云服务商(CSP):AWS、Azure、Google Cloud、Oracle Cloud 等,是最大采购方,用于内部 AI 服务和对外出租 GPU/TPU 实例。
- 大型 AI 实验室与企业:OpenAI、Anthropic、Meta、字节跳动等,自建或租用海量集群用于前沿模型训练。
- 服务器 OEM/ODM:超微、广达、纬颖、鸿海等,生产 GPU 服务器/整机柜,集成液冷与供电系统。
- 二次售卖市场与算力租赁平台:二手高端 GPU 流通活跃,小规模算力租赁商也构成零星市场。
关键指标
- 算力:FP16/BF16/FP8 TFLOPS 或 PFLOPS,以及对应稀疏算力(结构化稀疏可加倍)。
- 内存带宽:TB/s 级别,直接决定推理吞吐和某些训练阶段的效率。
- 容量:每个芯片/加速卡上的 HBM 总容量(GB),影响每卡可装载的模型大小。
- 互联带宽:片间通信带宽(如 NVLink 双向带宽),节点间网络带宽(如 400G/800G InfiniBand 或以太网)。
- 能效比:TOPS/W 或 PFLOPS/W,在数据中心功率密度限制下至关重要。
- 延迟:尤其是推理场景下,处理单个 token 的端到端延迟(毫秒级)。
- 扩展效率:当集群规模从 N 个加速器增至 2N 时,总吞吐量的提升比例(理想为 1.0,实际 0.7-0.9)。
供需与市场数据
根据多家研究机构汇总(如 TrendForce、IDC、650 Group 等,具体数字因付费壁垒不准确,以下为行业趋势定性):
- 全球云端 AI 芯片市场规模(含 GPU、ASIC、FPGA 等)近两年呈现爆发式增长,年增长率超 50%,2024 年估计已达数百亿美元级别。
- 供应严重受制于台积电 CoWoS 封装产能与 HBM 产能。主要厂商已预订未来数个季度的全部高端封装产能,交货周期长达数月。
- 需求端,训练侧受大模型军备竞赛驱动,推理侧仍未大规模爆发,但随着端侧 agent 化、生成式 AI 服务普及,推理芯片需求预期将超过训练,且更看重低功耗、低成本。
- 公开行业报告通常认为英伟达在云端 AI GPU 收入口径中处于领先位置,但具体份额会随统计范围、时间窗口和 GPU/ASIC 划分方式变化;自研 ASIC 在推理和云厂内部训练中的占比需要按云厂披露逐项核验。
- 地缘政治因素使得中国市场的芯片获取受限,国产替代(升腾、海光、寒武纪)在云端市场拿到一定份额,但硬件代差和软件生态仍是瓶颈。
代表公司与资本映射
- 英伟达 (NVIDIA):H100/H200/B200 系列,CUDA 生态护城河极深,即将推出基于 Blackwell 架构的 B100/B200 和 GB200 Superchip。公司市值一度破万亿美金,是 AI 基础设施的“定价锚”。
- 谷歌 (Google/Alphabet):TPU 系列已至 TPU v5p,用于内部搜索、广告、Gemini 模型训练及 Google Cloud 对外出租。
- 亚马逊 (Amazon):Trainium2 和 Inferentia2 系列,通过 AWS 提供,性价比优于 GPU 实例,主要用于自家及客户大模型训练推理。
- AMD:MI300X 系列,对标 H100/H200,主打大内存容量和开放式 ROCm 软件栈,正在积极争取大型客户。
- 英特尔 (Intel):Gaudi 3 加速器,继承自 Habana Labs,强调以太网互联和性价比,计划与 Falcon Shores 合并路线。
- 微软 (Microsoft):部署 Maia 100 加速器,用于 Azure 内部 AI 负载。
- Meta:公开 MTIA 系列芯片,现阶段主要针对推荐系统推理,未来可能扩展训练。
- Cerebras:CS-2/CS-3 系统,晶圆级引擎,单块芯片有数十万 AI 核心和巨大片上 SRAM,可直接单芯片训练大模型。
- 寒武纪、海光信息、华为昇腾 等中国厂商,遭受出口管制后,仍在迭代产品,满足国内市场需求。
产业影响与风险
-
训练算力需求:只要 scaling law 持续,对训练算力的需求就会保持高强度。GPU、HBM、先进封装和高速网络的收入弹性需要结合订单、交付、价格和毛利率分开验证;若模型训练效率提升过快或训练范式变革,需求增速可能阶段性放缓。
-
推理爆发论的悖论:AI 应用渗透后,推理算力长期可能超过训练。但推理芯片要求极低 TCO,毛利率相对训练芯片较低,且更容易被云厂自研 ASIC 侵蚀。因此需要分开观察云厂自研芯片、第三方加速器、网络、内存和边缘推理迁移到云端的不同路径。
-
配套环节:除芯片本身外,先进封装、HBM、液冷、高速交换机 (InfiniBand/以太网)、光模块等环节也会受云端 AI 集群建设影响。供给紧张需要用产能、良率、交期和客户订单验证,不能直接等同于业绩确定性。
-
生态标准变化:NVLink 的封闭生态正被 UALink、Ultra Ethernet 等联盟挑战。若能形成事实标准,可能降低 AI 芯片行业的准入门槛,但影响需要通过系统兼容性、软件栈成熟度和实际部署规模验证。
-
地缘与替代路径:出口管制会改变中国市场的供应选择和国产 AI 芯片迭代节奏。相关判断需要同时考虑性能差距、软件生态、流片风险、客户验证周期和实际采购口径。
常见误读纠偏
-
误读一:“AI 芯片就是看 TFLOPS,谁高谁厉害”
实际中,标称 TFLOPS 在真实工作负载下能达到多少是另一回事。内存带宽、软件编译器优化、多卡互联效率甚至比裸算力更重要。一颗 GPU 的软硬件协同可能使实际吞吐量比纸面算力只有其 70% 的 ASIC 还高。 -
误读二:“MoE 模型的激活参数就是总参数除以专家数,所以计算量小很多”
MoE 模型的“激活参数”通常等于共享层参数加上路由到的 top-K 专家的参数。这部分确实远小于总参数,但通信开销(All-to-All)大大增加,并且不能简单用总参数来估算算力需求。一定要区分“总参数量”和“每次前向的激活参数量”,后者决定单个 token 的计算 FLOPs,但对内存和系统通信提出更高的要求。 -
误读三:“有了 AI 专用卡,CPU 就不重要了”
在训练集群中,CPU 负责数据预处理、I/O、参数服务器的部分逻辑、以及启动控制信号等。大规模训练单卡可搭配十几个逻辑 CPU 核心。而且 PCIe/CXL 等协议仍需 CPU 根复杂。不能忽视整体异构计算系统的协同。 -
误读四:“云厂自研芯片能马上取代英伟达”
自研 ASIC 虽然硬件成本低,但软件生态的完备与开发者习惯的培养需要数年,大规模部署中的稳定性、运维工具链成熟度往往不及通用 GPU 平台。云厂自研芯片当前更多是补充与议价筹码,全面替代短期内不可能。
学习路径
- 入门:阅读“AI 芯片的基本原理与架构”综述文章,理解 MAC 阵列、存储层次、数据复用概念。观看英伟达、谷歌等公司的公开 Hot Chips 演讲,建立直观认知。
- 深入软件栈:亲手使用 PyTorch/tensorflow 编写简单网络并分析 CUDA kernel 或 XLA 编译图,理解算子下发与融合过程。了解 NCCL 的集合通信原语(Broadcast, AllReduce, ReduceScatter, All-to-All)。
- 研究系统级挑战:阅读 Megatron-LM、DeepSpeed 等分布式训练框架的论文,重点理解张量并行、流水线并行、ZeRO 优化器的显存优化原理。研究 MoE 通信优化(DeepSpeed-MoE 等)。
- 硬件与架构建模:用简单的性能模型(roofline model)分析给定网络在特定硬件上的计算/带宽瓶颈。调研先进封装、HBM 内部结构、SerDes 技术趋势。
- 追踪产业与投资逻辑:定期阅读 TrendForce、DIGITIMES、Next Platform 等媒体,跟踪大厂新品、产能动态、初创公司进展。结合代工厂财报分析供需。
一句话总结
云端 AI 芯片的终极形态不是一颗最强大的芯片,而是一个由先进封装、致命内存带宽、弹性高带宽互联和久经考验软件所构成的、可以线性扩展的“计算织物”。
延伸阅读与来源
- [未充分获取,典型参考源]:英伟达、AMD、英特尔等公司年度 GPU/加速卡技术白皮书和 GTC/Hot Chips 演讲资料。
- [行业研究机构]:TrendForce, Semico Research, IDC, 650 Group 等发布的 AI 芯片市场预测和先进封装产能报告(通常收费)。
- [学术与开源资源]:NVIDIA CUDA 编程手册、NCCL 文档;Google TPU 论文《A Domain-Specific Supercomputer for Training Deep Neural Networks》;Megatron-LM 论文;FlashAttention 论文。
- [新闻报道与博客]:Next Platform, SemiAnalysis, AnandTech, ServeTheHome 对新一代芯片架构的深度分析。
- [财务披露]:英伟达(Q 季报)、台积电、三星的营收构成与技术展望中,常有关于 HPC/AI 平台收入的定性与定量线索。