Core Group(DRAM 访问同步)

定义

Core group3D-Stacked AI Chip 上的一项硬件-软件协同优化:将物理相邻的 AI core 划分为若干组,组内通过 hardware request tracker 同步 DRAM 请求,减少多 core 异步访问共享 tensor 导致的 row-buffer 冲突

由 Voxel 论文(§4.4)提出,用于在增加 core 数量 scale FLOPS 时维持 DRAM 带宽利用率。

动机

3D AI chip 上单纯增加 core 数会:

  • 更多 core 访问共享 tensor shard → 访问 desync → 同一 DRAM bank 内 row 交错访问
  • Row-buffer 冲突加剧 → DRAM 带宽利用率下降 → memory-bound decode 性能受损

这与 §2.3 讨论的「多 core 数据共享导致 row conflict」直接相关;software-aware tensor-to-bank placement(§4.3)解决放置问题,core group 解决 并发访问时序 问题。

机制

Core Group (size = 8)
┌────┬────┬────┬────┐
│ C0 │ C1 │ C2 │ C3 │  ← 物理相邻 core
├────┼────┼────┼────┤
│ C4 │ C5 │ C6 │ C7 │
└────┴────┴────┴────┘
         │
    Hardware Request Tracker
    (同步组内 DRAM 请求 dispatch)
  • 组内 core 的 DRAM 请求经 request tracker 协调:在第 i 个请求被所有 core dispatch 完成前,不发起下一批请求
  • software-aware tensor-to-bank mapping 配合,可避免大部分不必要的 row conflict
  • 默认探索配置中 group size = 8(见 Voxel Simulator baseline)

实验结论(Voxel 论文)

场景发现
无 core group大 SA 或高 core count 设计 decode 性能差;SA 与 DRAM 利用率均低(Figure 15)
有 core group所有 core 配置均改善;core 数越高收益越大
1024 core, group size 8 vs 1LLM decode 最高 57% 加速
group size > 8额外收益可忽略;8 为 sweet spot
与 core scaling 协同适当 group size 使 3D chip 可更有效 scale core 数(Figure 14d)

Takeaway D2:scale core count 时,用简单 hardware logic(request tracker)同步同组 memory access;8 core 一组 可显著改善端到端性能与 DRAM 利用率。

与其他优化的关系

优化解决的问题层级
Software-aware tensor-to-bank mappingtensor 放置导致的 conflictCompiler / placement
Core group多 core 并发访问时序导致的 conflictHardware + 局部同步
Compute-shiftNoC 与 compute 重叠Compute paradigm
Dimension-ordered mappingNoC hop 最小化Tile-to-core mapping

Core group 不替代 tensor placement,二者互补。

适用工作负载

  • Decode / memory-bound:收益最大(DRAM 利用率是瓶颈)
  • Prefill / compute-bound:core group 非主优化路径;prefill 更依赖 compute-shift 与 NoC mapping

Prefill-Decode Resource Divergence

相关

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)