03 存储与内存层级洞察
- 章节编号:03
- 所属层:H 硬件层(与 02 章数据流强耦合;随 RTL/封装冻结代价高)
- 关联 ADR:ADR-009(片上 SRAM 容量/层级)、ADR-010(主存 LPDDR vs HBM)、ADR-011(存算/近存策略)、ADR-012(KV-cache/内存管理硬件支持)
- 上游依赖:01 章(内存/带宽是真瓶颈、量化后权重 2–4GB、KV-cache + 扩散迭代访存密集);02 章(数据流靠大 SRAM 降带宽、成熟工艺 + 软硬协同拿能效)
学习目标
- 前置知识:读过 01 章(端侧负载画像:VLA 权重量化后 2–4GB、KV-cache 与扩散迭代访存密集)与 02 章(数据流如何靠片上复用降带宽);知道「内存带宽」与「峰值算力」是两个独立指标;了解一次矩阵乘的浮点运算量约为
2·M·N·K。无需 RTL 或封装工艺背景。 - 学完产出:① 能画出端侧 SoC 的三级内存层次(寄存器/RF → 片上 SRAM(L1/全局)→ LPDDR5X 主存),说清每一级的带宽、容量、能耗量级与「离计算单元越远越慢越费电」的规律;② 能用 Roofline「内存墙」 一句话讲清「为什么 VLA 单次 token 解码是访存受限(memory-bound)而非算力受限」,并亲手算出「模型字节数 ÷ 带宽 = 单次访存下限」;③ 能对比 LPDDR5X 统一内存 vs HBM 在端侧的容量/带宽/功耗/成本取舍,说清为什么 Jetson Thor 选 128GB LPDDR5X 而非 HBM;④ 能解释 片上 SRAM / KV-cache 驻留 如何直接压低 π0 类流匹配动作头的 chunk 延迟;⑤ 能读懂本章的候选方案矩阵与四条 ADR,理解「容量-带宽-成本」三角为何锁死端侧选型。
- 阅读姿势:盯住一条主线——「端侧具身推理的性能上限,往往不是算力烧不动,而是数据搬不动」。无论是选 LPDDR 还是堆 SRAM、还是把 KV-cache 钉在片上,本质都在解决同一堵墙:DRAM 访问的带宽与能耗,比乘加运算贵一到两个数量级。读完请回答:「同一个 π0 动作专家,为什么把它塞进片上 SRAM 就能把 chunk 延迟压下来」。
1. 范围与目标
内存层级常是端侧推理真正的瓶颈与能耗大头(DRAM 访问能耗远高 于计算)。本章决定片上 SRAM 层级与容量、主存类型(LPDDR/HBM)、是否做存算/近存、以及面向 VLA 的 KV-cache 与内存管理硬件支持。
核心问题
- 片上 SRAM 要多大、几级层级?在 SRAM 工艺缩放停滞下如何权衡面积/成本?
- 主存选 LPDDR 还是 HBM?容量/带宽/功耗/成本如何取舍?
- 存算一体(CIM)/近存计算是否纳入?成熟度与风险?
- 如何在硬件层支持 KV-cache、动态 shape、多模型并发的内存管理?
2. 需求洞察(承接 01/02 章)
- 具体场景/任务:端侧人形,batch≈1,运行 OpenVLA 7B / π0 3.3B(流匹配 5–10 步,前缀 KV-cache)+ 感知 + 1kHz 控制。
- 由负载反推:
- 容量:量化后权重 ~2–4GB(当前)→ 3–8GB+(未来);叠加 KV-cache + 多模型 + 感知缓冲 → 主存 ≥8–16GB(ADR-003)。
- 带宽:Attention/KV-cache、扩散多步迭代访存密集 → ≥100–300 GB/s;对标 Thor 273 GB/s、Tesla AI5 0.77–1.5 TB/s。
- 能效:DRAM 访问能耗是计算的~100–200×(Eyeriss/学界)→ 必须靠大片上 SRAM + 高复用数据流(02 章 RS)把数据"留在片上"。
- 确定性:1kHz 控制路径访存须时序可预测 → 倾向关键数据片上驻留(避免 DRAM 抖动)。
- 端/边/中心差异:端侧 LPDDR + 适度 SRAM;中心 可上 HBM 拼带宽/容量。
2.1 内存金字塔:一张图看清「离计算越远越慢越费电」
端侧 SoC 的存储不是一块平地,而是一座金字塔:越靠近乘加单元(MAC)的存储越快、越省电,但越小越贵;越往外走容量越大、越便宜,但带宽和能耗急剧劣化。这座金字塔的形状,决定了后面所有的架构取舍。
读这张图的关键:金字塔顶端的寄存器访问能耗记作 1×,那么一次 LPDDR 访问的能耗是它的 100–200 倍(Eyeriss 数据)。这意味着——同一份数据,若能少一次 DRAM 往返、多留在片上复用一次,省下的不只是时间,更是电池。这条规律是 02 章「大 SRAM + 高复用数据流」和本章一切选型的物理根因。