AI芯片内功心法大全:为什么没有一款芯片能通吃所有AI任务?
2026 年 3 月,AI 教育者 Avi Chawla 在 X(原 Twitter)上发布了一条爆款帖子——一张非常硬核的“AI芯片内功心法图”。
这张图正好解释了:当代码(比如大模型推理)跑到最底层的硬件时,不同的芯片内部到底是怎么干活的。
1. CPU(中央处理器):全能的“大管家”
图中特征:
Control(控制单元)很大,有很多ALU(算术逻辑单元)但数量不多。
有很深的L1/L2/L3 Cache,最后才是DRAM。
标注了Scalar + SIMD (latency-optimized)。
核心思想:样样通,样样松,但绝不偏科(通用性最强)。
内部逻辑:
CPU的核心是“控制单元”,它擅长处理复杂的逻辑、分支判断(比如
if-else、操作系统调度)。它只有几个强力核心,每个核心都配了超多缓存(Cache),目的是减少访问慢速内存(DRAM)的次数,追求低延迟。
它不适合干重复的累活(比如几十亿次乘法),但擅长发号施令。
在AI中的角色:指挥官。负责调度GPU/TPU干活,处理数据预处理和逻辑控制。
2. GPU(图形处理器):疯狂的“搬砖工”
图中特征:
SIMT Parallel Processing(单指令多线程并行)。
有Streaming Multiprocessor(流式多处理器),里面有大量的CUDA Core。
有L1/L2 Cache,最后是GDDR/HBM。
核心思想:人多力量大,简单重复的工作我来。
内部逻辑:
GPU把几千个简单的核心(CUDA Core)塞进芯片。它们不像CPU那样各自为战,而是听同一个指令(SIMT),同时算不同的数据。
它是为图形渲染(画三角形)和深度学习(矩阵乘法)量身定制的。
它有高带宽内存(HBM),但依然依赖片外内存,调度开销相对较大。
在AI中的角色:主力军。负责训练大模型、做科学计算,算力爆炸但功耗也高。
3. TPU(张量处理器):极致的“矩阵工厂”
图中特征:
Tensor/Matrix Processing。
中间有个巨大的Systolic Array(脉动阵列),全是MAC(乘加器)。
有Data Buffers,最后才是HBM。
核心思想:为了算矩阵,我连脑子都不要了。
内部逻辑:
TPU是Google发明的专用ASIC。它放弃了通用性,把所有的晶体管都用来做矩阵乘法(GEMM)。
Systolic Array(脉动阵列)是精髓:数据像流水线一样在阵列里流动(权重从左边进,激活值从上面进),每个单元只做乘法和加法,结果传给下一个。这极大减少了去内存拿数据的次数。
它是编译器驱动的(XLA),执行路径在编译时就定死了,非常确定。
在AI中的角色:超级计算器。专门用于神经网络中密集的线性代数运算,训练/推理效率极高。
4. NPU(神经处理单元):省电的“边缘管家”
图中特征:
Neural Inference Compute。
有Neural Compute Engine(很多MAC)。
有Tensor Buffer,最后连Clip/System Memory(通常是低速的系统内存)。
核心思想:我在手机里,我要省电!
内部逻辑:
NPU专为低功耗推理设计(如手机、摄像头、IoT设备)。
它不像GPU那样追求峰值算力,而是追求能效比(TOPS/W)。
内部大量使用低精度计算(如INT8)和片上SRAM,尽量减少访问外部DDR内存(因为内存访问最耗电)。
在AI中的角色:端侧小助手。让你的手机能本地做人脸识别、语音助手,不用联网。
5. LPU(语言处理单元):确定的“Token打印机”
图中特征:
Deterministic Streaming Compute(确定性流式计算)。
Functional Slices(功能切片),有MXM, VXM, SXM等模块。
巨大的On-chip SRAM (220 MB)。
关键标注:No off-chip DRAM in critical path(关键路径无片外内存)。
核心思想:为了快,我把所有东西都塞进肚子里,绝不卡壳。
内部逻辑:
LPU(如Groq的芯片)是最新物种,专为大模型推理(LLM)优化。
它彻底取消了片外DRAM(HBM),把模型权重全部放在220MB的巨大片上SRAM里。这意味着数据读取时间恒定(Zero Cache Miss)。
Deterministic(确定性):所有操作由编译器提前排好队,运行时像流水线一样精准同步,没有缓存未命中和动态调度的开销。
它是软件优先(Software-First)设计的,编译器把一切都规划好了。
在AI中的角色:极速打字机。能实现每秒几百个Token的推理速度,专治大模型“嘴笨”。
架构 | 核心哲学 | 关键结构 | 内存策略 | 典型场景 |
|---|---|---|---|---|
CPU | 通用 & 低延迟 | 强控制单元 + 深缓存 | 依赖多级Cache + DRAM | 系统控制、逻辑调度 |
GPU | 大规模并行 | 数千CUDA Core (SIMT) | HBM/ GDDR (高带宽) | 模型训练、图形渲染 |
TPU | 极致矩阵运算 | Systolic Array (脉动阵列) | HBM (编译器调度) | 云端密集计算 |
NPU | 低功耗推理 | Neural Engine + 低精度MAC | 片上SRAM + 低速系统内存 | 手机/IoT边缘AI |
LPU | 确定性 & 零延迟 | Functional Slices + 超大SRAM | 全片上SRAM (无片外) | 大模型实时推理 |
CPU/GPU还在努力克服“内存墙”(计算快,拿数据慢),所以它们有大Cache。
TPU/NPU/LPU则是“破罐子破摔”:既然拿数据慢,那我就把数据塞到计算单元旁边(SRAM),或者让数据自己流动(脉动阵列),甚至干脆不要外部内存(LPU)。
概念 | 内部萌芽 | 对外公开 | 大规模商用/出圈 |
|---|---|---|---|
TPU | 2013(Google 立项) | 2016年5月(I/O 大会) | 2017 Cloud TPU |
LPU | 2016(Groq 提出) | 2020 前后(芯片推出) | 2023-2024 GroqCloud + 2026 英伟达 |
一、CPU"不擅长AI"的真相:不是不能算,是"单位能量不划算"
很多人误以为CPU"算不动矩阵",其实CPU单核算力并不弱,AVX-512一次能算16个float32,问题在于:
1. 算力密度太低
CPUdie上大部分面积给了Cache和控制逻辑(你之前学的L1/L2/L3),真正的ALU占比小
GPU相反:80%以上的die都是计算单元,Cache很小→ 同样的硅面积,GPU的MAC数量是CPU的100倍以上
2. 内存带宽被"卡脖子"
AI计算的本质是:输出 = 权重 × 输入 + 偏置,一次矩阵乘要:
从内存读权重(几十GB)
读激活值
算完写回
CPU的DDR4带宽~50GB/s,GPU的HBM~1TB/s,差20倍。CPU算力再强,数据喂不上来也是"饿死"。
3. 能耗比悬殊
CPU算1次FP32 MAC:~1pJ(皮焦)
GPU算1次FP16 MAC:~0.1pJ
NPU算1次INT8 MAC:~0.01pJ
同样算1万亿次乘加,CPU可能要100W,NPU只要1W——手机电池扛不住CPU跑AI
所以CPU不是"不能算AI",是"算AI太贵太耗电"。这就是专用硬件出现的根本经济动因。
二、"内存墙":AI芯片架构分化的真正导火索
你材料里提到了Cache Miss,但没点透——"内存墙"(Memory Wall)才是过去10年AI芯片创新的最大驱动力。
冯·诺依曼瓶颈在AI时代彻底爆发
计算单元:每18个月翻倍(摩尔定律)
内存带宽:每10年才翻3倍
→ 差距越拉越大
到2020年代:
NVIDIA H100的算力:2000 TFLOPS
HBM带宽:3TB/s
理论算完1个字节数据能做的运算:2000T / 3T ≈ 667次FLOP/Byte
但实际AI模型每个权重只被用几次就废弃→ 算力严重闲置,等数据等到花儿谢。
五种芯片的"破墙"策略各不相同
架构 | 破墙思路 | 代价 |
|---|---|---|
GPU | 堆HBM带宽+Tensor Core | 功耗700W+,贵 |
TPU | Systolic Array让数据"流动"不回头 | 只能算矩阵 |
NPU | 片上SRAM+INT8,少访问外存 | 只能推理 |
LPU | 彻底取消外存,全放片上SRAM | 单芯片容量小,要组网 |
CPU | 大Cache硬扛 | 扛不住AI规模 |
三、为什么是"5种"而不是"1种通吃"?场景决定架构
专用芯片的悖论:越专用,越高效,但越窄用。AI任务谱系太广,单一架构无法通吃:
训练 vs 推理:性质完全不同
训练:需要高精度(FP32/FP16)、需要反向传播、需要超大算力→GPU/TPU主场
推理:可低精度(INT8/FP8)、顺序生成token、追求时延→NPU/LPU主场
云端 vs 边缘:功耗预算天差地别
云端:电力不限→可以700W狂奔→GPU/H100
手机:电池3W预算→必须NPU,INT8+SRAM
批量 vs 实时:吞吐量vs时延
批量推理(如翻译1000句话):要吞吐→GPU/TPU
实时对话(token-by-token):要确定性低时延→LPU
这就是为什么市场分化出5种架构——每种都是对特定约束的最优解。
四、异构计算的协同:未来系统的真实样子
你材料最后提到了"异构计算",我展开讲一个真实的AI请求是怎么在5种芯片间流转的:
以"手机上问AI NPC一个问题"为例
1. 麦克风采集音频 → NPU跑ASR(语音转文字,本地,低功耗)
2. 文字请求决策:
- 简单问题(天气/闹钟)→ NPU本地跑小LLM(3B参数)
- 复杂问题(剧情推理)→ 压缩后上传云端
3. 云端接收:
- CPU编排任务,分发
- GPU/TPU跑大模型推理(批量)
- 若是实时对话 → LPU集群跑(确定性低时延)
4. 生成结果回传手机
5. NPU跑TTS(文字转语音)→ 扬声器播放
全程5种芯片协同,用户无感知,但背后是:
2次NPU(端侧语音+小模型)
1次GPU/TPU(云端训练好的模型)
1次LPU(如果走Groq这类实时推理)
CPU全程编排
开发者要懂的"硬件感知"
模型量化:FP32→FP16→INT8,让模型能在NPU/LPU上跑
算子融合:减少内存往返(如Conv+BN+ReLU合成一个kernel)
编译调度:XLA/TVM/MLIR把模型映射到目标硬件
动态卸载:小模型端侧NPU,大模型云端GPU/LPU
这就是为什么"AI工程师"现在必须懂硬件——不懂硬件的AI优化,就是盲调超参。
和你之前所有知识的终极闭环
把从"晶体管"到"AI芯片"的全链路串起来:
AI需要专用硬件,根本不是"CPU算不动",而是"内存墙+能耗比"双重挤压下的经济选择:GPU用HBM堆带宽、TPU用脉动阵列让数据流动、NPU用INT8+SRAM省电、LPU用全片上SRAM消延迟——5种架构对应5类约束的最优解,未来是CPU编排+多芯片协同的异构计算时代,开发者必须"硬件感知"才能把模型跑得又快又省~
AI 硬件的演进:CPU、GPU、TPU、NPU 与 LPU 全面对比(技术科普) | 工业智能算网
