Cheat Sheet速查表
跳转到主要内容

现代 AI 芯片与算力架构全景指南:从微架构机理、PD 分离到集群工程选型

深度拆解现代 AI 数据中心算力芯片架构机理:剖析 CPU、GPU、TPU、LPU、NPU、DPU 以及华为昇腾、AWS Trainium、Meta MTIA、微软 Maia 底层硬件原理、Roofline 访存受限推导、PD 分离架构、集群互联拓扑、选型决策树与实战调优三大法则。

在生成式大语言模型(LLM)与多模态智能体(Agent)全面渗透各行各业的背景下,计算系统的核心矛盾已经从“单核主频不足”演变为“大规模张量并行吞吐与微秒级生成时延的物理极限博弈”。算力硬件不再是孤立的单卡芯片,而是由多种异构处理器、超高带宽片上网络、高密度存储介质与分布式无损织网所构成的超级计算集群。

我们在 现代 AI 芯片与数据中心算力架构速查表 中提供了高密度的参数矩阵、显存计算尺与运维命令速查。本文将进一步系统剖析 CPU、GPU、TPU、LPU、NPU、DPU 六大主流计算架构 及 华为昇腾、AWS Trainium、Meta MTIA、微软 Maia 等国内外智算版图 的底层微架构机理、Roofline 物理受限公式、大模型 Prefill/Decode 分离范式,并给出生产级工程选型决策树与三大调优法则。


现代 AI 算力微架构与数据中心流水线全景图

现代 AI 芯片微架构与 Roofline 物理分化全景图


第一模块:大模型算力瓶颈与数学机理(Roofline 模型)

评估任何 AI 芯片的算力瓶颈与理论性能上限,首先必须理解计算机体系结构经典的 Roofline 模型(屋顶线模型)。

1. 算力强度与性能上限推导

定义某个算子或模型阶段的算力强度(Arithmetic / Operational Intensity) I:

I=总浮点运算次数 (FLOPs)总访存数据量 (Bytes)(单位: FLOP/Byte)I = \frac{\text{总浮点运算次数 (FLOPs)}}{\text{总访存数据量 (Bytes)}} \quad (\text{单位: FLOP/Byte})

系统可达到的最大实际计算性能 P 受到硬件峰值算力 P_peak 与硬件内存物理带宽 B_mem 的共同制约:

P=min⁡(Ppeak,  I×Bmem)P = \min\left( P_{\text{peak}}, \; I \times B_{\text{mem}} \right)
性能 P (TFLOPS)
  ^
  |        算力受限区 (Compute-Bound)
P_peak +---------------------------------- (平台区: 算力上限)
  |   /
  |  /  访存受限区 (Memory-Bound)
  | /   (斜率 = 内存物理带宽 B_mem)
  |/
  +-------------------------------------> 算力强度 I (FLOP/Byte)
      I_knee (转折点)
  • 转折点(Knee Point):
    Iknee=PpeakBmemI_{\text{knee}} = \frac{P_{\text{peak}}}{B_{\text{mem}}}
    - 若 `I < I_knee`,程序落入**访存受限区(Memory-Bound)**,处理器的计算核心大部分时间在饥饿等待数据从内存读入,此时堆砌算力核心(TFLOPS)毫无意义,系统性能直接与内存物理带宽 `B_mem` 成正比。 - 若 `I >= I_knee`,程序落入**算力受限区(Compute-Bound)**,内存带宽不再是主要瓶颈,性能由芯片的峰值运算吞吐 `P_peak` 决定。

2. Prefill 与 Decode 阶段的物理本质彻底分化

在大语言模型推理中,Prefill 与 Decode 两个阶段的计算特征有着天壤之别:

A. 提示词预填阶段(Prefill Phase)

用户输入由 S 个 Token 组成的 Prompt。自注意力机制与前馈全连接层(FFN)对这 S 个 Token 同时进行并行矩阵乘法计算:

FLOPs≈2×Pparam×S,Iprefill≈2×Pparam×S2×Pparam=S(FLOP/Byte)\text{FLOPs} \approx 2 \times P_{\text{param}} \times S, \qquad I_{\text{prefill}} \approx \frac{2 \times P_{\text{param}} \times S}{2 \times P_{\text{param}}} = S \quad (\text{FLOP/Byte})

此时模型权重仅需从显存中加载一次(大小约为 2 × P_param 字节),但用于并行处理全部 S 个 Token。当 Prompt 长度达到数千乃至数万时,I_prefill >> I_knee,系统稳稳处于**计算密集型(Compute-Bound)**区域。此时需要极高密度的 Tensor Core 并行浮点算力(如 NVIDIA H100 的 FP8/FP16 张量核心、Google TPU 脉动阵列或华为昇腾 3D Cube)。

B. 自回归解码阶段(Decode Phase)

模型在生成阶段是逐 Token 自回归进行的。每生成一个新 Token(Batch Size 为 1 时),都需要将整个大模型的全部参数权重矩阵从显存中完整扫描一遍:

单步 FLOPs≈2×Pparam×1,Idecode≈2×Pparam×12×Pparam≈1(FLOP/Byte)\text{单步 FLOPs} \approx 2 \times P_{\text{param}} \times 1, \qquad I_{\text{decode}} \approx \frac{2 \times P_{\text{param}} \times 1}{2 \times P_{\text{param}}} \approx 1 \quad (\text{FLOP/Byte})

以 NVIDIA H100(SXM5)为例:

  • 峰值 FP16 张量算力 P_peak ≈ 989 TFLOPS
  • HBM3 显存带宽 B_mem ≈ 3.35 TB/s
  • 硬件拐点 I_knee = (989 × 10^12) / (3.35 × 10^12) ≈ 295 FLOP/Byte

由于 I_decode ≈ 1 << 295,在 Decode 阶段,H100 昂贵的 Tensor Core 理论利用率不足 1%!芯片完全处于显存带宽饥饿状态(Memory-Bound)。

这是推动行业发展 LPU(基于超大带宽 SRAM) 以及大显存带宽芯片(如配备 8 TB/s HBM3e 的 NVIDIA B200)的核心数学动因。


第二模块:核心计算芯片微架构深度剖析

flowchart TD
    ROOT["现代算力处理器全景分类"]
    
    ROOT --> GEN["通用控制中枢"]
    ROOT --> PAR["通用并行加速"]
    ROOT --> DSA["特定领域专用架构 (DSA)"]
    ROOT --> INFRA["基础设施与互联卸载"]

    GEN --> CPU["CPU (中央处理器)<br/>复杂控制流 / 分支预测 / 多级缓存"]
    PAR --> GPU["GPU (图形处理器)<br/>SIMT 架构 / Tensor Core / HBM 堆叠"]
    
    DSA --> TPU["TPU (谷歌张量处理器)<br/>二维脉动阵列 / OCS 光网络"]
    DSA --> ASCEND["昇腾 NPU (华为达芬奇)<br/>3D Cube / CANN 软件栈 / HCCS"]
    DSA --> LPU["LPU (语言处理器 Groq)<br/>全片上 SRAM / 确定性编译器"]
    DSA --> CLOUD_ASIC["云厂商专用 ASIC<br/>AWS Trainium / Meta MTIA / 微软 Maia"]
    DSA --> EDGE_NPU["端侧 NPU<br/>Apple ANE / 高通 Hexagon / Intel NPU"]
    
    INFRA --> DPU["DPU (数据处理器)<br/>RDMA / 网络线速解包 / 安全虚拟化"]

1. CPU:复杂的乱序执行与多级缓存体系

通用 CPU(如 Intel Xeon 与 AMD EPYC)的微架构哲学是最小化单线程任务的处理延迟。

  • 超标量乱序执行(Out-of-Order Execution, OoO):CPU 投入了巨额晶体管面积用于动态分析指令依赖关系,预测分支跳转并乱序调度执行指令,以便在任何未知代码逻辑下保持流水线饱满。
  • 庞大的多级 Cache 层次:L1/L2/L3 缓存占据了 CPU 核心的大量面积,用以掩盖主内存访问的高延迟(60~100 ns)。
  • 现代大模型角色:
    • 调度总控:大语言模型应用不仅是神经网络前向推理,还包含大量复杂的工具调用(Tool Calling)、上下文检索逻辑(RAG)、正则表达式匹配、JSON 序列化与反序列化。这些复杂的条件分支是 CPU 的绝对统治领域。
    • 统一内存与 CXL:现代数据中心 CPU 能够直接编址管理数 TB 乃至数十 TB 的系统内存与 CXL 内存池,适合作为大模型海量知识图谱和高维向量数据库的主控节点。

2. GPU:SIMT 架构与专用 Tensor Core

现代 GPU(以 NVIDIA Hopper/Blackwell 与 AMD Instinct 系列为代表)的微架构哲学是最大化浮点运算的聚合吞吐量。

  • SIMT(Single Instruction, Multiple Threads)架构:GPU 将成千上万个计算单元组织成流多处理器(Streaming Multiprocessors, SM)。32 个线程组成一个线程束(Warp),共享同一个指令发射器,以极致的面积效率执行相同的计算指令。
  • Tensor Core 张量核心:摒弃传统逐点标量算术逻辑单元(ALU),直接在硬件层面实现矩阵乘累加(MMA, Matrix Multiply-Accumulate)指令:
    D=A×B+CD = A \times B + C
    单时钟周期内即可完成 4 × 4 或 16 × 16 小矩阵的整块乘加,算力密度比通用矢量单元提升一个数量级。
  • HBM 高带宽内存(High Bandwidth Memory):通过硅通孔(TSV)技术将多层 DRAM 晶圆垂直堆叠,并利用微凸块(Micro-bumps)直接焊接在硅中介层(Silicon Interposer)上,以数千条引脚提供高达 3~8 TB/s 的物理带宽,是支撑大模型训练的唯一成熟介质。

3. 华为昇腾:DaVinci 达芬奇架构与 CANN 软件生态

华为昇腾(Ascend)系列是国内应用最为广泛的本土自研 AI 算力基座:

  • 3D Cube 达芬奇架构:
    • 核心计算单元由 Cube 矩阵计算单元、Vector 矢量计算单元 与 Scalar 标量计算单元 组成;
    • Cube 单元专精 GEMM 密集矩阵乘加,单周期可完成 $16 \times 16 \times 16$ 的半精度乘加运算;Vector 单元负责激活函数(Softmax、GELU)、LayerNorm 等非矩阵计算;Scalar 单元负责基本控制循环。
  • HCCS(Huawei Cache Coherence System):
    • 昇腾 8 卡服务器内部通过自研 HCCS 高速总线全互联,提供类似于 NVIDIA NVLink 的硬件级缓存一致性通信,保证张量并行(TP)的高速扩展。
  • CANN 异构计算软件栈:
    • 向上支持主流 PyTorch(通过 torch_npu 插件),向下直接调度昇腾芯片内部计算流,配合 MindSpeed 加速库实现主流千亿模型(LLaMA、DeepSeek、Qwen)的高效预训练与微调。

4. TPU:脉动阵列(Systolic Array)与光交换网络

Google TPU(Tensor Processing Unit)是特定领域架构(DSA)在大规模机器学习领域的开山代表。

  • 脉动阵列数据流驱动:
    • 传统 GPU 在每一步计算中,操作数和中间结果频繁读写本地寄存器或共享内存;
    • TPU 的核心矩阵乘法单元(MXU)设计为 $128 \times 128$ 或 $256 \times 256$ 的二维处理单元网格。数据从左侧以波状脉动注入,权重从顶部预加载,中间累加和像心脏跳动一样在相邻的处理单元(PE)之间直接流动传递;
    • 这种设计彻底避免了海量中间结果对寄存器堆的读写功耗,将硅片面积与能耗几乎 100% 倾注在矩阵算力上。
  • 光交换机(Optical Circuit Switch, OCS)拓扑:
    • 在 TPU v4/v5p 规模的 Pod 集群中,Google 利用微机电系统(MEMS)反射镜构建了全光网络拓扑;
    • 集群拓扑可在毫秒级完全软件定义重构(如从 3D Torus 动态变换为环形或网格),并能在某块芯片或光纤损坏时毫秒级自动绕行,保证了万卡预训练集群极高的平均无故障时间(MTBF)。

5. LPU:静态确定性编译与全片上 SRAM

以 Groq 为代表的语言处理单元(Language Processing Unit, LPU)基于其独创的张量流处理器(Tensor Streaming Processor, TSP)架构,旨在彻底解决 LLM 自回归 Decode 阶段的显存带宽瓶颈。

  • 全片上 SRAM 架构:
    • 彻底舍弃片外 HBM 或 DRAM。芯片内全部存储空间均采用超高速度的静态随机存取内存(SRAM);
    • SRAM 单芯片容量虽小(单芯片仅约 230 MB),但片上物理带宽高达 80 TB/s(是顶级 HBM3e 的 10 倍以上),且访问延迟降至亚纳秒级;
    • 面对 ( I \approx 1 \text{ FLOP/Byte} ) 的 Decode 阶段,80 TB/s 的带宽直接将单卡每秒可扫描的参数量推向极致,单流生成速率轻松突破数百 Tokens/s。
  • 纯静态确定性编译器驱动:
    • LPU 硬件内部完全剔除了传统芯片的动态分支预测器、硬件乱序调度器和非确定性缓存命中机制;
    • 整个计算流水线中的每一个时钟周期、每个数据在硅片空间上的流动位置,均在编译期由静态编译器完全计算排定;
    • 这保证了极短且完全可预测的生成时延(Zero Jitter),适合对时延抖动极其敏感的高并发在线系统。
  • 工业部署妥协:由于单芯片 SRAM 仅数百 MB,运行一个 70B 模型(FP16 下约 140 GB 权重)必须通过高速网络将数十至数百张 LPU 级联为一个分布式集群共同提供存储容量。

6. 云厂商定制 ASIC:AWS Trainium/Inferentia、Meta MTIA 与微软 Maia

北美科技巨头为了摆脱对单一 GPU 供应链的依赖并削减云端 TCO,纷纷自研专属 AI 加速器:

  • AWS Trainium 2:采用 NeuronCore-v2 架构,通过 NeuronLink-v2 构建万卡集群,专为万亿参数混合专家模型(MoE)的跨节点预训练优化;AWS Inferentia 2 则以超高片上 Cache 专攻云端吞吐推理。
  • Meta MTIA:采用 RISC-V 控制核心搭配专用稠密/稀疏张量引擎,专门针对 Meta 核心收益支柱——推荐系统(DLRM)与内部 Llama 模型推理量身定制。
  • 微软 Azure Maia 100:台积电 5nm 工艺,专为 OpenAI 大模型训练与推理打造,集成定制以太网协议与机柜闭环直接芯片液冷(Direct-to-Chip Liquid Cooling)。

7. DPU:现代 AI 超级数据中心的基础设施卸载

在千卡与万卡规模的智算中心内,通信开销直接决定了模型训练与并行推理的扩展效率。DPU(数据处理单元)扮演着“数据中心操作系统硬件加速器”的核心角色。

  • 网络协议栈硬件卸载:DPU 硬件实现全套 RDMA(远程直接数据存取)协议栈与 RoCEv2 拥塞控制逻辑(如 DCQCN、TIMELY),网络数据包直接绕过操作系统内核与主 CPU,以硬件线速(400 Gbps ~ 800 Gbps)写入目标主机的 GPU 显存(GPUDirect RDMA);
  • 存储与安全虚拟化解耦:现代训练任务需周期性保存数百 GB 的 Checkpoint 权重。DPU 硬件加速 NVMe-oF(基于以太网或光纤网的 NVMe 存储解耦),并在线提供线速硬件 AES/TLS 数据加解密,彻底免除计算卡被 I/O 阻塞的困扰。

第三模块:超大规模 AI 集群高速互联与网络拓扑

当模型参数达到千亿乃至万亿级别时,单张芯片甚至单个服务器节点无法承载。互联网络的拓扑结构与物理带宽成为决定算力集群扩展加速比的命门。

flowchart TD
    subgraph NODE["单个 AI 计算节点 (Scale-Up 纵向扩展域)"]
        GPU1["GPU 0"] <--> |NVLink 5 / HCCS 1.8 TB/s| NVSwitch["NVSwitch / HCCS 硬件交换总线"]
        GPU2["GPU 1"] <--> |NVLink 5 / HCCS 1.8 TB/s| NVSwitch
        GPU3["GPU 2"] <--> |NVLink 5 / HCCS 1.8 TB/s| NVSwitch
        GPU4["GPU 3"] <--> |NVLink 5 / HCCS 1.8 TB/s| NVSwitch
    end

    subgraph NETWORK["跨节点数据中心网络 (Scale-Out 横向扩展域)"]
        DPU_A["节点 A DPU"] <--> |InfiniBand NDR / RoCEv2 800G| SpineSwitch["脊叶式交换架构 (Spine-Leaf)"]
        DPU_B["节点 B DPU"] <--> |InfiniBand NDR / RoCEv2 800G| SpineSwitch
    end

    NVSwitch -.-> |PCIe 5.0 / 6.0| DPU_A

1. Scale-Up(节点内纵向扩展):微秒级高带宽通信域

在单机 8 卡或机柜级互联(如 NVIDIA NVL72、华为昇腾超节点)中,芯片间主要运行张量并行(Tensor Parallelism, TP)与专家并行(Expert Parallelism, EP),通信极其高频,要求 TB/s 级的双向带宽。

  • NVLink 与 NVSwitch:NVIDIA Blackwell 架构下的 NVLink 5 提供单卡 1.8 TB/s 双向互联带宽,借助机架内置的物理 NVSwitch 托盘,可将整机柜 72 块 GPU 虚拟化为一块具备统一显存寻址空间的“巨型单芯片”。
  • 华为 HCCS 与 Google ICI:昇腾节点利用 HCCS 总线实现全卡点对点高速直连;Google 则在 TPU Pod 内部采用芯片直连网格,配合光交换网络实现低延迟的跨芯片点对点全双工互联。

2. Scale-Out(节点间横向扩展):无损智算网络

在跨机架、跨机房的大规模扩展中,芯片间主要运行流水线并行(Pipeline Parallelism, PP)与数据并行(Data Parallelism, DP / ZeRO)。

  • InfiniBand(IB):专用无损网络标准,采用基于信用积分(Credit-Based)的硬件流控机制,交换机转发延迟仅为数十纳秒,完全消除数据包排队抖动与重传开销,是大模型预训练的事实标准网络。
  • RoCEv2(RDMA over Converged Ethernet):在传统通用以太网基础设施上通过 PFC(优先级流控)与 ECN(显式拥塞通知)技术模拟无损通道,具备成本优势,常用于具备规模运维能力的超大规模公有云厂商。

第四模块:现代 AI 算力工程选型决策树

在面对实际生产场景时,可参考以下工程选型决策路径:

flowchart TD
    START["🎯 AI 算力选型起点"] --> Q1{"核心任务类型为何?"}

    Q1 --> |超大模型预训练 / 全量微调| TRAIN["🔥 训练任务"]
    Q1 --> |生产级部署与在线推理| INFER["⚡ 推理任务"]
    Q1 --> |系统编排 / 数据流水线| ORCH["🛠️ 基础设施与编排"]

    TRAIN --> T_SCALE{"参数量与供应链约束?"}
    T_SCALE --> |国际主流云端 / 万亿参数 MoE| T1["首选: NVIDIA H100/H200/B200 集群<br/>+ InfiniBand 网络 + DPU<br/>(或 Google TPU v5p Pods)"]
    T_SCALE --> |自主可控与国产化替代| T2["首选: 华为昇腾 910B/910C 集群<br/>+ CANN 软件栈 + HCCS 互联"]
    T_SCALE --> |< 14B 中小模型轻量微调| T3["单机 8 卡 H100 / L40S<br/>或 AMD MI300X 高显存单机节点"]

    INFER --> I_ENV{"部署环境为何?"}
    I_ENV --> |端侧手机 / PC / 边缘嵌入式| I_EDGE["端侧 NPU (Apple ANE / Qualcomm / Intel)<br/>+ INT4/AWQ 紧凑量化"]
    I_ENV --> |云端公有云 / 数据中心服务| I_CLOUD{"推理阶段瓶颈重心为何?"}

    I_CLOUD --> |超长 Prompt 预填 / 大批量文档抽取| P1["GPU (H100/H200) / TPU v5e / 昇腾 910B<br/>(高 TFLOPS 算力密度)"]
    I_CLOUD --> |毫秒级极速交互 / 语音实时流式解码| P2["PD 分离架构: GPU 预填 + LPU/B200 解码<br/>(利用超高带宽 SRAM / HBM3e)"]
    I_CLOUD --> |高性价比多租户高并发吞吐| P3["标准 GPU 节点 + vLLM / TensorRT-LLM<br/>(FP8 量化 + PagedAttention 动态批处理)"]

    ORCH --> O1["网络与安全卸载: 部署 DPU (BlueField-3 / Nitro)<br/>业务编排与 Agent 循环: 部署高主频多核 CPU (EPYC / Xeon / 倚天710)"]

第五模块:工业级算力优化三大实战调优法则

法则一:PD 分离架构(Prefill-Decode Disaggregation)

在大模型单节点推理系统中,Prefill 阶段的突发大计算量会严重阻塞同时进行的 Decode 逐 Token 生成,导致系统首字时延(TTFT)与中间生成时延(ITL, Inter-Token Latency)产生巨大抖动。

  • 实施方案:
    1. 建立专用的 Prefill 节点池:采用算力密集型芯片(如高 TFLOPS 的 GPU、TPU 或昇腾 910B),负责快速消化长上下文输入并生成 KV-Cache;
    2. 建立专用的 Decode 节点池:采用显存带宽极大或配置了全片上 SRAM 的芯片(如 LPU 或 B200),专门负责高效拉取 KV-Cache 并流式自回归吐字;
    3. 节点间通过 高速 RDMA 网络(基于 DPU) 进行微秒级 KV-Cache 内存传输,实现算力与带宽资源的彻底解耦。

法则二:显存带宽瓶颈破解(KV-Cache 压缩与量化感知)

由于自回归生成本质上是访存受限过程,优化系统吞吐的关键在于减少单步需搬运的数据体积。

  • PagedAttention 显存零碎片化:传统连续内存分配会导致 60%80% 的 KV-Cache 显存浪费。借鉴虚拟内存分页机制,将动态变化的 KV-Cache 分割为离散物理块,可直接将并发 Batch 吞吐提升 24 倍。
  • 量化感知权衡:
    • FP8 格式:目前 Hopper/Blackwell 与最新开源框架(vLLM)的事实标准,相比 FP16 吞吐提升约 2 倍,几乎无精度损失;
    • INT4 / AWQ / GPTQ 权重只读量化(Weight-Only Quantization):在 Decode 阶段将模型权重压缩 4 倍,极大削减显存总带宽需求,使消费级或边缘芯片亦能流畅运行千亿参数模型。

法则三:拓扑感知通信调度(Topology-Aware Parallelism)

在大规模并行训练与多卡推理时,严禁盲目均匀切分模型。必须使通信频率与物理互联带宽严格匹配:

  • 张量并行(Tensor Parallelism, TP):每层 Transformer 内部均发生 All-Reduce 通信,必须限制在单机内走超高带宽 NVLink / HCCS 域;
  • 流水线并行(Pipeline Parallelism, PP):仅在各 Layer 边界传输激活值,通信量相对较小,适合跨节点走 InfiniBand / RoCEv2 网络;
  • 数据并行(Data Parallelism, DP):采用分片优化器状态(ZeRO / FSDP),通信模式为周期性的 All-Gather 与 Reduce-Scatter,结合 DPU 的网络卸载引擎实现计算与通信的双缓冲完全重叠(Overlapping)。
← 返回博客浏览速查表 →