Graphcore IPU (Intelligence Processing Unit)

Graphcore 的 AI 加速器产品线,以 Colossus 架构为核心:大量简单 core 全互联,每 core 配本地 SRAM,形成分布式内存空间而非 HBM 统一寻址。Hot Chips 2021 披露的 Colossus Mk2 IPU 是 Voxel 论文选用的 silicon 验证平台。

Colossus Mk2 关键规格

参数数值
AI core 数1,472(全互联)
片内互连总带宽7.8 TB/s
分布式 SRAM 总容量896 MB
分布式 SRAM 聚合带宽62 TB/s
架构特点每 core 本地 SRAM;core 间经 NoC 访问远端 SRAM

单颗 operator 的 tensor 可完整放入片上分布式内存,适合 emulate 3D AI chip 的 distributed DRAM bank 访问模式。

与 wiki 中其他加速器对比

维度Graphcore IPU Mk2Cerebras WSEGroq LPU
Core 规模1,472~900K PE256/chip
内存模型分布式 per-core SRAM分布式片上 SRAM片上 SRAM
互连全互联 NoC24-color meshC2C plesiosynchronous
典型用途训练/推理(Poplar)大模型训练/推理低延迟 decode

Cerebras Wse 同属 多 core + 分布式内存 范式,但 IPU 规模更小、更适合作 research emulator。

Voxel 论文中的验证角色

尚无商用 3D-stacked AI chip,Voxel 作者在 IPU Mk2 上构建 hardware emulator

960 core  → 模拟 3D AI chip 的 AI core
512 core  → 模拟分布式 DRAM bank(数据存于对应 core SRAM)

验证流程:

  1. Emulated Time:在 IPU 上执行,数据从 SRAM 加载(无 DRAM latency)
  2. Emulated Time + DRAM Latencies:replay IPU trace,叠加 Ramulator DRAM 延迟
  3. Voxel Simulated Time:Voxel 端到端仿真

Llama2-13B、Gemma2-27B、OPT-30B、Llama3-70B、DiT-XL 上,Voxel 与带 DRAM 延迟的 emulator 误差 0.24%–6.8%。纯 SRAM emulator 平均快 12.7%(SRAM 任意 access pattern 均可满带宽)。

详见 Voxel SimulatorVoxel 3D-Stacked AI Chip LLM Inference

相关页面

Citations

[1] [raw/papers/Exploring the efficiency of 3D-stacked AI chip architecture for LLM inference with voxel.pdf](raw/papers/Exploring the efficiency of 3D-stacked AI chip architecture for LLM inference with voxel.pdf) [2] Simon Knowles, “Graphcore Colossus Mk2 IPU,” IEEE Hot Chips 33, 2021.