AI计算范式迁移与芯片架构的范式转移
2023年生成式AI的爆发让AI算力成为全球科技竞争的焦点:训练一个70B参数的大语言模型需要数千块高性能芯片,单次推理的延迟要求从秒级压缩到毫秒级,端侧AI应用的功耗限制被死死卡在1W以内。在这种极端的需求下,我们观察到同一个深度学习工作负载在不同架构的芯片上运行时,性能能差3-4个数量级,能效差4-5个数量级:用通用CPU跑ResNet-50推理的吞吐量仅为10张/秒,能效不足0.005张/焦耳;而用专用推理加速器跑同样的任务,吞吐量可达2000张/秒,能效超过100张/焦耳,差距超过2万倍。这种巨大的性能鸿沟并非来自工艺制程的差异——三类芯片往往采用相同的7nm、5nm工艺——其本质是计算范式的迁移倒逼芯片架构发生范式转移:当工作负载从串行通用计算转向并行计算、再转向张量密集的深度学习计算时,芯片架构也必须从通用CPU转向并行GPU,最终转向深度适配深度学习特征的专用AI加速器。
---
1.1 通用计算到并行计算:CPU与GPU的架构演进逻辑
芯片架构的每一次范式转移都和工作负载的核心特征变化直接绑定。在深度学习革命爆发前的半个多世纪里,通用计算的工作负载以串行任务为主:办公软件、数据库、科学计算的串行逻辑、分支判断、随机访存占据了绝大多数计算量,这类任务的核心特征是数据依赖强、并行度低、精度要求高,直接决定了CPU架构的设计逻辑。
1.1.1 冯诺依曼架构与CPU的通用计算范式
1945年冯诺依曼提出的存储程序架构奠定了现代通用处理器的设计基础:计算单元、控制单元、存储单元分离,通过总线传输数据。为了最大化串行任务的执行效率,CPU在几十年的演进中形成了三个核心设计特征: 1. 复杂的控制单元:包含分支预测器、乱序执行引擎、寄存器重命名单元,用于最大化指令级并行(ILP),将串行任务中的潜在并行指令挖掘出来并行执行,将IPC(每时钟周期执行指令数)从最初的0.x提升到当前x86/ARM旗舰CPU的3-4。 2. 多级缓存 hierarchy:L1/L2/L3缓存的容量占CPU核心面积的50%以上,用于降低随机访存的延迟,匹配通用工作负载的访存局部性。 3. Few Big Cores 设计:旗舰CPU的核心数通常在8-64核之间,每个核心的面积在10mm²以上,优先保证单核性能,而非核心数量。
这种设计在通用计算场景下效率极高,但面对深度学习工作负载时却暴露了根本性的缺陷:深度学习核心的张量运算(矩阵乘法、卷积)是高度规则、无数据依赖、可大规模并行的计算,CPU的复杂控制单元、小核心数量、低带宽的内存架构完全无法匹配这类负载的特征。以Intel旗舰服务器CPU Xeon Platinum 8480+为例,其56核的FP16峰值算力仅为1.5 TFLOPS,满载功耗350W,能效仅为0.0043 TFLOPS/W,跑ResNet-50推理的吞吐量仅为10张/秒,连实时视频分析的需求都难以满足。
1.1.2 GPU的并行计算范式转移
CPU的局限性在2000年前后随着3D图形渲染的需求爆发变得尤为突出:3D图形的像素渲染、顶点变换、光照计算都是典型的数据并行任务,每个像素、每个顶点的计算逻辑完全一致,没有数据依赖,可以用成千上万的简单计算单元同时处理。这种需求直接催生了GPU的并行计算架构,其核心设计逻辑和CPU完全相反: 1. Many Small Cores 设计:以NVIDIA A100为例,其包含108个流多处理器(SM),每个SM包含128个CUDA Core,总计算单元数达6912个,每个CUDA Core的面积仅为CPU核心的1/100,控制逻辑极简,没有乱序执行、分支预测等复杂单元。 2. 高带宽内存架构:GPU采用GDDR6X或HBM显存,显存带宽可达1.5-2TB/s,是CPU内存带宽的5-10倍,匹配并行计算的高带宽需求。 3. SIMT(单指令多线程)执行模型:一组线程(Warp,通常32个线程)执行同一条指令,无需分支预测,大幅降低了控制逻辑的功耗。
2007年NVIDIA发布CUDA编程框架,第一次将GPU从图形处理器变成了通用并行计算平台,标志着计算范式从串行通用计算正式转向并行通用计算。GPU的并行架构相比CPU有10-100倍的并行算力提升,能效提升5-10倍,很快成为深度学习研究的标配硬件。2012年AlexNet在ImageNet竞赛中以15.4%的错误率夺冠,比第二名低了10.9个百分点,引爆了深度学习革命,而AlexNet的训练正是用两块GTX 580 GPU在6天内完成的——如果用当时的CPU集群训练,需要数月甚至数年时间。
但GPU本质仍是通用并行处理器,其架构仍然存在大量为图形渲染设计的冗余单元:比如纹理单元、光栅化单元(ROP)占GPU芯片面积的15%-20%,在跑深度学习任务时完全闲置;GPU的调度单元、缓存结构是为通用并行任务设计的,并未针对张量运算的访存特征优化,内存带宽利用率仅为30%-50%。这些冗余单元和未优化的架构导致GPU的能效相比专用加速器仍有5-10倍的提升空间,无法满足大模型时代对算力成本和能效的极致要求。
---
1.2 深度学习工作负载的核心计算特征
专用AI加速器的诞生并非偶然,而是深度学习工作负载的核心特征和通用处理器架构的不匹配性达到临界点的必然结果。深度学习工作负载有三个无可替代的核心特征,直接决定了专用加速器的设计方向。
1.2.1 高并行度:张量运算的天然并行性
深度学习的核心运算是张量运算,尤其是矩阵乘法、卷积运算,这类运算的每个输出元素的计算都是独立的,没有任何数据依赖,天然支持大规模并行。以Transformer模型的注意力层为例,其核心运算是查询矩阵$Q \in \mathbb{R}^{B \times S \times H}$和键矩阵$K \in \mathbb{R}^{B \times S \times H}$的矩阵乘法($B$为batch size,$S$为序列长度,$H$为隐藏层维度),得到注意力分数矩阵$A \in \mathbb{R}^{B \times S \times S}$,总计算量为$O(B \cdot S^2 \cdot H)$。当$B=32$、$S=2048$、$H=8192$时,单层注意力层的计算量高达69 billion FLOPs,其中每个元素的乘加运算都可以并行执行,没有任何先后依赖。
这种高并行度意味着芯片架构不需要复杂的控制逻辑来挖掘指令级并行,只需要尽可能