LPU Architecture(Language Processing Unit)

LPU 是 Groq(现 NVIDIA)设计的推理专用加速器架构。核心思想:不追求峰值算力,而是追求可预测的低延迟。

设计哲学

  • SRAM-first:片上 SRAM 作为主要工作存储,不用硬件缓存
  • 向量为主:320-byte 固定向量作为计算/内存/通信的统一工作单元
  • 编译器控制:显式调度一切,无运行时决策
  • 确定性互联:C2C 链路 + plesiosynchronous 协议

Slice 架构(ISCA 2020 原始设计)

LPU 将架构重组为单功能单元组 “slice”,slice 间通过 streaming registers 和 scratchpad SRAM 传递数据:

Slice功能
MXM (Matrix)密集 MAC,矩阵乘法
VXM (Vector)逐元素运算、类型转换、激活函数
SXM (Switch)排列、旋转、分发、转置
MEM片上 SRAM 存储
  • Slice 水平排布:数据水平流
  • Slice 内指令垂直 pump:指令垂直穿越功能单元
  • 概念上类似 systolic array:指令垂直流、数据水平流

SRAM 与内存层次

层级GPULPU
L1Hardware-managed cacheStreaming registers
L2Hardware-managed cacheScratchpad SRAM(编译器管理)
HBM高带宽外部内存无(SRAM-only,LP30 为 500 MB)
DDR5Host DRAMFPGA 附加 DRAM(LPX 中最多 256 GB/FPGA)

关键权衡:SRAM 极快(低延迟、高带宽)但密度低、成本高 → LPU 的 TTFT 和 tokens/sec/user 极快,但总吞吐受 SRAM 容量限制(weight 占满后 KV cache 空间不足)。

芯片世代对比

世代制程SRAM算力状态
LPU Gen 1GF 14nm230 MB750 TFLOPs (INT8)量产(2020)
LPU Gen 2Samsung SF4X失败(112G SerDes 问题)
LP30 (Gen 3)Samsung SF4500 MB1.2 PFLOPs (FP8)NVIDIA 产品化中
LP35Samsung SF4500 MB规划中(+NVFP4)
LP40TSMC N3P + CoWoS-RHybrid bonded DRAMFeynman 世代,NVLink 协议

与 GPU 的核心区别

维度GPULPU
内存层次HBM + 缓存(硬件管理)SRAM(编译器管理)
调度硬件 warp scheduler编译器静态调度
数据搬运隐式(缓存一致性)显式(编译器编排)
优化目标峰值吞吐可预测延迟
通信NVLink(自适应路由)C2C(确定性)
适用负载Prefill、attention(stateful)FFN/MoE expert(stateless)

相关页面

Citations

[1] nvidia-groq3-lpx-blog-2026-04.md [2] [raw/articles/GTC 2026 – The Inference Kingdom Expands.md](raw/articles/GTC 2026 – The Inference Kingdom Expands.md)