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

现代 AI 芯片与数据中心算力架构速查表:CPU、GPU、TPU、LPU、NPU、DPU 与国内外算力全景选型

在生成式大语言模型(LLM)与多模态智能体工业化落地的时代,算力基础设施的选型直接决定了系统的训练收敛速度、推理首字时延(TTFT)、每 Token 生成成本与集群能效比。现代 AI 计算已从单一的“GPU 单点包打天下”全面走向**异构计算(Heterogeneous Computing)与特定领域架构(DSA, Domain-Specific Architecture)**协同编排体系。

本速查表彻底剔除过时型号与非技术营销水分,立足真实工业级芯片微架构与数据中心拓扑,高密度汇总 CPU、GPU、TPU、LPU、NPU、DPU 六大计算架构 及 国内外主流智算芯片版图(华为昇腾、寒武纪、海光、AWS Trainium、Meta MTIA、微软 Maia 等),并提供首屏核心架构图、大模型显存容量推导计算尺与集群运维诊断命令速查。

📖 深度微架构剖析、Roofline 模型访存受限推导与超大规模集群实战调优请参阅:现代 AI 芯片与算力架构全景指南:从微架构机理、PD 分离到集群工程选型

🔗 相关矩阵速查:大模型与智能体技术概念全景词典速查表 · AI 核心算法全景速查表 · 运维与高并发架构全景知识图谱速查表 · Linux 常用命令速查表


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

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


1. 现代 AI 数据中心端到端计算流水线(PD 分离技术路径)

大模型请求在现代 AI 云数据中心中的完整生命周期遵循标准的数据卸载 ➔ 控制编排 ➔ 预填计算 ➔ 解码生成 ➔ 响应分发链路:

步骤阶段名称核心执行芯片核心任务与硬件行为主要瓶颈模式典型代表硬件
01网络接入与数据包解包DPU (数据处理单元)接收客户端 TLS 加密流量,硬件线速解密;处理 VPC 虚拟网络路由与防火墙;通过 RDMA/RoCEv2 零拷贝写入主机内存I/O 与包处理线速 (PPS / Gbps)NVIDIA BlueField-3, AWS Nitro, AMD Pensando
02业务编排与上下文调度CPU (中央处理器)运行 API 网关与 Agent 智能体循环;执行 Tokenizer 分词、向量库 RAG 召回、Function Calling 参数验证与批处理组织单核主频与控制延迟 (IPC / Branch)AMD EPYC 9004/9005, Intel Xeon 5th/6th, NVIDIA Grace, 阿里倚天 710
03提示词预填 (Prefill)GPU / TPU / 专用ASIC一次性并行处理全部输入 Prompt Token,计算注意力矩阵,生成 KV-Cache;高密度 GEMM 矩阵乘法计算受限 (Compute-Bound, TFLOPS)NVIDIA H100/H200/B200, Google TPU v5p/v6e, 华为昇腾 910B/910C
04自回归逐字解码 (Decode)LPU / 高带宽 GPU基于上文 KV-Cache,逐个生成下一个 Token;每次生成必须读取全部模型权重;依赖超高带宽存储访存受限 (Memory-Bound, GB/s)Groq LPU (片上 SRAM 80TB/s), NVIDIA B200 (8 TB/s HBM3e)
05流式输出与连接回收DPU (数据处理单元)将流式 Token 实时封包为 SSE/gRPC 网络响应回传客户端;释放计算集群网络队列,彻底解除 CPU/GPU 的 I/O 阻塞网络中断与协议栈开销NVIDIA BlueField-3, AWS Nitro

2. 六大核心计算架构参数全景对照表

芯片类型 (英文全称)核心微架构与计算机制主流存储介质与典型带宽核心物理受限瓶颈典型 TDP 功耗核心编程生态 / 编译器工业界典型代表芯片工业级核心角色定位
CPU
(Central Processing Unit)
复杂多发射超标量、深度乱序执行 (OoO)、高级分支预测、大容量多级 Cache系统 DDR5 / LPDDR5X (100 ~ 500 GB/s);支持 CXL 扩展矩阵并行算力密度低、矢量单元受限150W ~ 500Wx86-64 / ARMv9 原生指令集, OpenMP, AVX-512, AMXIntel Xeon Emerald/Granite Rapids, AMD EPYC Turin, NVIDIA Grace, AWS Graviton4集群中枢:API 网关、工具调用、Agent 控制流、轻量数据流清洗
GPU
(Graphics Processing Unit)
SIMT (单指令多线程)、大规模流多处理器 (SM)、高密度专用 Tensor Cores片外堆叠 HBM3 / HBM3e (3.35 ~ 8.0 TB/s) 或 GDDR6/GDDR7显存带宽受限 (Decode阶段)、先进封装产能 (CoWoS)300W ~ 1000W+ (需风冷/液冷)NVIDIA CUDA, TensorRT, Triton, AMD ROCm, PyTorchNVIDIA Hopper (H100/H200), Blackwell (B200/GB200), AMD Instinct MI300X/MI325X行业事实标准:超大规模分布式预训练、主流大模型全量微调与高吞吐量推理
TPU
(Tensor Processing Unit)
二维脉动阵列 (Systolic Array) 硬件矩阵乘加、确定性数据流驱动、自研 OCS 光交换拓扑片外高带宽 HBM (HBM2e / HBM3, 1.6 ~ 4.8 TB/s)深度绑定专有生态、非标算子定制开发门槛高300W ~ 600WOpenXLA, JAX, PyTorch-XLA (深度绑定 Google Cloud)Google TPU v4, TPU v5p (训练旗舰), TPU v5e (高性价比), TPU v6e (Trillium)云端专属矩阵引擎:Google 内部产品超大规模集群训练 (Gemini) 与云上租户吞吐
LPU
(Language Processing Unit)
纯张量流处理器 (TSP),静态确定性指令流水线,全片上超高带宽 SRAM,无外部 DRAM 延迟全片上高密度 SRAM (单片约 230 MB,片上带宽达 80 TB/s)片上存储容量极度受限(必须多卡级联存放模型)、不具备训练能力150W ~ 300W确定性编译器 Groq Compiler (静态编译模型计算图)Groq LPU (TSP 架构)极致低延迟解码器:专精大模型自回归 Decode 阶段,实现 300~500+ Tokens/s 实时交互
NPU
(Neural Processing Unit)
针对神经网络轻量算子剪裁的专用 DSA 架构,支持 INT8/INT4/FP8 低精度定点与浮点混合计算统一内存 (Unified Memory) 或共享 LPDDR5/LPDDR5X (50 ~ 150 GB/s)算力上限有限、不支持长上下文与大参数模型全量运行2W ~ 30W (高能效比 TOPS/W)Apple CoreML, Qualcomm QNN, Intel OpenVINO, ONNX RuntimeApple Neural Engine (ANE), Qualcomm Hexagon, Intel NPU, AMD XDNA 2端侧与边缘引擎:手机、AI PC、嵌入式设备本地离线轻量模型与 Copilot 辅助加速
DPU / IPU
(Data / Infrastructure Processing Unit)
多核高效能 CPU 集群 (如 Arm Cortex-A 系列) + 可编程网络加速引擎 (ASIC/P4) + 硬件密码学加速器板载专用 LPDDR5 / DDR4 (独立于主机内存),配硬件 DMA 引擎纯业务矩阵计算能力弱(不参与神经网络张量运算)50W ~ 150WDOCA (NVIDIA), Linux 内核驱动, P4 编程语言, Open vSwitchNVIDIA BlueField-3, AMD Pensando Elba/Pollara, AWS Nitro, Intel IPU E2000数据中心基座:RoCEv2/RDMA 传输加速、VPC 隔离、NVMe-oF 存储解耦与安全防线

3. 国内外智算芯片生态版图速查

在全球与本土算力多元化的工业背景下,除了传统的 NVIDIA 与 AMD 之外,各大云厂商与国产芯片厂商已建立起完备的自主智算产品线:

A. 国产自主算力体系(中国智算核心阵列)

厂商 / 品牌代表芯片型号核心微架构 / 算力规格显存介质与带宽互联协议与技术核心软件栈与框架适配工业落地定位与适用场景
华为昇腾
(Huawei Ascend)
Ascend 910B / 910C
(推理: Ascend 310P)
自研达芬奇 DaVinci 架构 (3D Cube 矩阵计算 + Vector 矢量单元);910B 约 320 TFLOPS FP1664GB ~ 96GB HBM2e (1.2 ~ 1.6 TB/s)HCCS (片间高速互联,类似 NVLink) + PCIe 5.0CANN 异构计算底座、MindSpore、PyTorch 昇腾适配插件 (torch_npu)、MindSpeed国内大模型主力替代底座:千亿模型预训练、全量微调、政企私有化部署首选
寒武纪
(Cambricon)
思元 370 (MLU370)
思元 590 (MLU590)
智能芯片专用微架构,采用 Chiplet 模块化拼叠技术;MLU590 专为大模型预训练优化24GB LPDDR5 (MLU370) / 64GB HBM (MLU590, ~1.8 TB/s)MLU-Link 芯片互联技术Cambricon Neuware 软件栈、BANG 编程语言、PyTorch/TensorFlow 适配工业视觉加速、智能云服务器、中大模型训练与推理集群
海光信息
(Hygon)
海光 DCU
(深算一号 / 二号 / 三号)
兼容通用 GPGPU 架构,全精度浮点计算能力完备,高度兼容主流 GPU 生态16GB ~ 64GB HBM2 / HBM2e专用高速总线互联接口DTK (Developer Toolkit),源码级兼容 ROCm / HIP 生态,主流开源框架开箱迁移科学计算、中长尾模型微调、金融量化交易与大数据分析
摩尔线程
(Moore Threads)
MTT S4000
(夸娥万卡集群)
MUSA (Moore Threads Unified System Architecture) 全功能 GPU 架构48GB GDDR6 (768 GB/s)MT-Link 专有互联总线 (双向 240 GB/s)MUSA 软件栈、MUSIFY 代码自动迁移转换工具、Torch-MUSA千卡/万卡级大模型预训练与推理集群、图形渲染与视频编码混合应用
昆仑芯
(Kunlunxin)
昆仑芯 2 代 / 3 代自研 XPU 架构,针对矩阵与矢量加速专门剪裁的高性能计算单元16GB ~ 64GB HBM / GDDR6XPU-Link 高速卡间互联百度飞桨 (PaddlePaddle) 深度优化、昆仑芯 XPU-SDK、PyTorch 适配百度搜索/推荐主力加速、公有云 AI 推理服务、智能驾驶路侧计算

B. 北美头部云巨头自研 ASIC 矩阵

厂商芯片名称微架构设计哲学显存与互联规格软件栈与驱动支持工业级核心价值
AWS
(Amazon)
Trainium 2
(训练专用)
自研 NeuronCore-v2 架构,专注万亿参数 MoE 模型极大规模分布式预训练32GB ~ 96GB HBM,配 NeuronLink-v2 全互联拓扑 (可扩展至 10 万卡集群)AWS Neuron SDK (深度绑定 PyTorch 与 AWS 基础服务)极大幅度降低 AWS 云上分布式模型预训练的基础设施成本
AWS
(Amazon)
Inferentia 2
(推理专用)
低成本大模型推理 DSA,每个芯片集成高密度计算引擎与超大片上高速 Cache片上 Cache + 32GB LPDDR5 (带宽成本显著优于 HBM)AWS Neuron SDK为云端大规模在线 API 推理服务提供极致的每瓦性能与每请求成本
MetaMTIA v1 / v2
(自研加速器)
混合架构(RISC-V 控制核心 + 专用密集/稀疏张量矩阵阵列),专攻 Meta 核心业务板载 LPDDR5 内存 + PCIe 5.0 扩展自研软硬件编译器,深度绑定 PyTorch 2.0 内部编译链路承载 Instagram / Facebook 推荐排序(DLRM)与 Llama 内部部署
微软
(Microsoft)
Azure Maia 100针对 OpenAI GPT 系列大模型计算特征深度定制的 5nm 专用 AI 加速芯片高规格 HBM 堆叠,采用定制以太网通信网络与闭环机柜液冷Azure AI 基础设施软件栈,深度适配 Triton 与 DeepSpeed为 Azure OpenAI 服务提供专有底层算力支撑,摆脱单点供应链依赖

4. 大模型显存容量与带宽实战计算尺

在模型选型、卡数规划与推理部署前,必须通过数学公式精算显存占用:

1. 核心数学计算公式

A. 模型静态权重显存占用

Mweight=P×Bits8×(1+Overhead)(单位: GB)M_{\text{weight}} = P \times \frac{\text{Bits}}{8} \times (1 + \text{Overhead})\quad (\text{单位: GB})
  • P:模型总参数量(例如 70B = 700 亿参数)。
  • Bits:量化位宽(FP16/BF16 为 16,FP8 为 8,INT4 为 4)。
  • Overhead:框架层显存碎片与非权重张量冗余(通常估算取 1.15 ~ 1.20)。

B. 动态 KV-Cache 显存占用(单并发每 Token 开销)

在采用分组查询注意力(GQA, Grouped-Query Attention)的现代 Transformer 中:

Mkv_per_token=2×L×nkv×dhead×BytesPerParam(Bytes/Token)M_{\text{kv\_per\_token}} = 2 \times L \times n_{\text{kv}} \times d_{\text{head}} \times \text{BytesPerParam}\quad (\text{Bytes/Token})
  • L:模型 Transformer 层数(如 Llama-3-70B 中 L = 80)。
  • n_kv:KV 注意力头数(如 Llama-3-70B 中 n_kv = 8)。
  • d_head:单注意力头维度(通常为 128)。
  • BytesPerParam:每个参数字节数(FP16 为 2,FP8 为 1)。
  • 系数 2:分别保存 Key 向量与 Value 向量。

总 KV-Cache 显存随并发批尺寸 B 与序列长度 S 线性增长:

Mkv_total=B×S×Mkv_per_token(Bytes)M_{\text{kv\_total}} = B \times S \times M_{\text{kv\_per\_token}}\quad (\text{Bytes})

2. 主流大模型显存占用速查表

下表给出了纯模型权重显存(含 15% 框架冗余)与每 1k 上下文产生的 KV-Cache 大小对照:

模型规制隐藏层数 (L)KV 头数 (n_kv)FP16 / BF16 权重显存FP8 权重显存INT4 / AWQ 权重显存单序列 4k 上下文 KV-Cache (FP16)单序列 32k 上下文 KV-Cache (FP16)
7B ~ 8B (Llama-3-8B)328~16 GB~8.5 GB~4.8 GB~0.5 GB~4.0 GB
14B (Qwen-2.5-14B)488~31 GB~16.5 GB~9.2 GB~0.75 GB~6.0 GB
32B (Qwen-2.5-32B)648~70 GB~37 GB~21 GB~1.0 GB~8.0 GB
70B (Llama-3-70B)808~155 GB~82 GB~45 GB~1.3 GB~10.5 GB
405B (Llama-3-405B)1268~900 GB~470 GB~255 GB~2.0 GB~16.2 GB

💡 显存避坑定则:当并发批尺寸 B = 64、上下文为 32k 时,单 70B 模型的 KV-Cache 占用高达 64 × 10.5 GB = 672 GB,是模型自身权重的 4 倍以上!必须启用 PagedAttention 与 FP8 KV-Cache 压缩。


5. 智算集群运维与状态监控速查矩阵

在大规模集群环境中,快速定位芯片掉卡、降频、NVLink 错误与内存瓶颈至关重要:

1. NVIDIA GPU 常用监控与诊断命令

# 1. 实时以 CSV 格式输出 GPU 核心状态(索引、名称、温度、GPU 利用率、显存使用、实时功耗)
nvidia-smi --query-gpu=index,name,temperature.gpu,utilization.gpu,utilization.memory,memory.used,memory.total,power.draw --format=csv -l 1

# 2. 查看机器内 NVLink 硬件链路状态与连接健康度
nvidia-smi nvlink -s

# 3. 检查 NVLink 链路误码率与 CRC 错误包计数(若持续增加说明线缆或插槽物理松动)
nvidia-smi nvlink -e

# 4. 查看当前节点多卡之间的物理拓扑关系(识别 NVLink / PCIe / 跨 NUMA 节点状态)
nvidia-smi topo -m

# 5. 锁定 GPU 核心与显存频率(基准压测与防时钟抖动)
sudo nvidia-smi --lock-gpu-clocks=2100,2100

# 6. 使用 NVIDIA DCGM 运行整机 3 级深度硬件稳定性巡检(包含显存压力测试)
dcgmi diag -r 3

2. 华为昇腾 NPU (CANN) 常用监控与诊断命令

# 1. 查询当前系统全部昇腾芯片的基础状态、HBM 显存占用与驱动固件版本
npu-smi info

# 2. 实时刷新当前昇腾芯片算力利用率、温度与功率消耗(以 1 秒为间隔)
npu-smi info -w 1

# 3. 检查昇腾芯片间 HCCS 高速互联拓扑状态(相当于 NVLink 拓扑检查)
npu-smi info -t hccs -i 0

# 4. 查看昇腾主板温度传感器与电源供电健康度
npu-smi info -t board -i 0

# 5. 抓取昇腾设备错误代码与历史硬件告警日志
npu-smi info -t error-code -i 0

3. AMD Instinct GPU (ROCm) 常用监控命令

# 1. 查看 MI300X 显存占用、利用率、温度与功耗
rocm-smi --showuse --showmeminfo vram --showtemp --showpower

# 2. 显示 AMD 节点内 XGMI 卡间全互联总线拓扑
rocm-smi --showtopo

4. 模型算力利用率 (MFU, Model FLOPs Utilization) 生产级估算脚本

MFU 是衡量大模型训练和推理是否充分压榨芯片算力的核心指标(行业一流水平约为 40% ~ 55%):

def calculate_mfu(
    num_params_billion: float,      # 模型参数量 (单位: B, 如 70.0)
    throughput_tokens_per_sec: float,# 集群总体每秒生成 Token 吞吐量
    num_gpus: int,                  # 参与计算的总 GPU 卡数 (如 8)
    peak_tflops_per_gpu: float,     # 单卡峰值算力 (如 H100 FP16 峰值为 989.0)
    is_training: bool = False       # True 为训练 (6*P), False 为推理 (2*P)
) -> float:
    """计算模型算力利用率 (MFU, Model FLOPs Utilization)"""
    # 每次前向计算每个 Token 需 2*P FLOPs;反向传播需 4*P FLOPs;训练总计 6*P
    factor = 6.0 if is_training else 2.0
    flops_per_token = factor * (num_params_billion * 1e9)
    
    # 实际达成的计算吞吐 (TFLOPS)
    achieved_tflops = (throughput_tokens_per_sec * flops_per_token) / 1e12
    
    # 硬件理论峰值总算力 (TFLOPS)
    total_peak_tflops = num_gpus * peak_tflops_per_gpu
    
    mfu = (achieved_tflops / total_peak_tflops) * 100.0
    return round(mfu, 2)

# 示例: 8 张 H100 运行 70B 模型在线推理,集群吞吐为 800 Tokens/s
mfu = calculate_mfu(num_params_billion=70, throughput_tokens_per_sec=800, num_gpus=8, peak_tflops_per_gpu=989)
print(f"当前集群实测 MFU: {mfu}%")
# 若 MFU < 15%,说明系统严重处于访存瓶颈或网络通信阻塞状态

6. 核心架构认知总结与演进方向

  1. 特定领域架构(DSA)终结单一芯片垄断:算力正从“同构堆核”转向“异构协同”。CPU 负责灵活控制流,DPU 负责网络与安全卸载,GPU/TPU/昇腾专攻长上下文预填,LPU 专精低延迟自回归解码,NPU 渗透终端低功耗场景。
  2. PD 分离架构成为推理生产环境的终极范式:计算密集型的 Prefill 阶段与访存密集型的 Decode 阶段对硬件微架构的需求完全对立。通过跨节点高速 RDMA 网络分离部署,能避免资源挤占,将集群每瓦吞吐提升数倍。
  3. 软硬件协同设计(Co-Design)决定生命线:没有优秀编译器(如 TensorRT、Triton、CANN、OpenXLA)的专用芯片往往沦为纸面参数。算子覆盖率、自动算子融合与显存虚拟化调度能力是衡量芯片工程落地成熟度的最关键判据。