Skip to content

5. 核心模块 ​

本章逐个拆解 Meta AI 基础设施的核心组件,每个模块讲清楚它解决什么问题、怎么设计、关键参数。这些模块是 第 3 章 架构的实体化。

5.1 计算模块 ​

5.1.1 Grand Teton — OCP 开放 GPU 平台 ​

Grand Teton 是 Meta 自研、通过 OCP 开放的 GPU 平台,定位是整机柜级别的单体集成。

维度设计
集成GPU + CPU + 电源 + 控制 + 网络织物 → 单体机柜
动机减少机柜内线缆、简化供电与远程管理、可规模化复制
开放性设计通过 OCP 公开,供应链可围绕统一规范优化

5.1.2 Catalina — GB200 Blackwell 机柜 ​

Catalina 是与 NVIDIA 协同设计的 Blackwell(GB200)机柜,是 Meta 上液冷的代表。

维度参数
结构6-rack pod:中间 2 机柜装 72 个 Blackwell GPU
功耗~140 kW/pod(约为传统风冷机柜的 8×)
算力~360 PFLOPS FP16/pod
冷却4 个 AALC(Air-Assisted Liquid Cooling) 机柜(风液混合)
后续GB300 在规划中

5.1.3 MTIA — 自研训练/推理 ASIC ​

MTIA(Meta Training and Inference Accelerator)与 Broadcom 合作,遵循 "High velocity / Inference First / PyTorch Native" 三原则,cadence 约 6 个月一代。下表汇总四代(基于官方《Four MTIA Chips in Two Years》):

代次定位关键提升
MTIA 300ranking/recommendation 训练(R&R)内置 NIC chiplet + message engine + 近存 reduction;1 compute chiplet + 2 network chiplet + HBM;PE 含 2 个 RISC-V 向量核 + Dot Product Engine + SFU + Reduction Engine + DMA
MTIA 400推理FP8 FLOPS +400%、HBM 带宽 +51%(vs 300);2 compute chiplet;72 加速器 scale-up 域(switched backplane);MX8/MX4
MTIA 450推理(含 LLM)HBM 带宽 2×(vs 400);MX4 FLOPS +75%;硬件加速 attention/FFN(Softmax、FlashAttention);MX4 FLOPS 是 FP16 的 6×;2027 初量产
MTIA 500推理HBM 带宽 +50%、HBM 容量 +80%、MX4 FLOPS +43%(vs 450);2×2 compute chiplet + HBM + 2 network chiplet + SoC chiplet(PCIe);2027 量产

从 MTIA 300 到 500:HBM 带宽 4.5×、计算 FLOPS 25×(MX8→MX4)。400/450/500 共用同一 chassis/rack/network(drop-in 升级)。

MTIA 已有论文:ISCA'23(MTIA 100)、ISCA'25(MTIA 200,原名 MTIA 1 / MTIA 2i)与 ISCA'26(MTIA 300)。MTIA 已大规模部署服务广告(ads)推理。

MTIA 300 深度拆解(2026-08-24 官方专文 + ISCA'26) ​

2026-08-24 Meta 发表 MTIA 300 专文(engineering.fb.com),并配以 ISCA'26 论文,给出大量硬数字。MTIA 300 是 Meta 第一颗专为训练设计的芯片(ranking/recommendation 模型),2026 年初已量产上线,其核心创新是把网络与通信卸载做进芯片:

维度设计
内置 NIC2 个 network chiplet 内含 12 个自研 800 Gbps Ethernet RDMA NIC,合计 1.2 TB/s I/O 带宽且完全不经过 PCIe——消除了 GPU 架构中 host-device-NIC 的瓶颈路径
网络灵活性同一 NIC 池软件可配为 Scale-Up(16 节点机柜内最高 1 TB/s)或 Scale-Out(跨机柜 200 GB/s),换模式不改硬件
通信卸载16 个 message engine(各含 RISC-V 核 + NIC 接口 + 近存计算块)独立于计算网格执行 AllReduce / AllGather 等集合通信,归约吞吐 2.8 TB/s;并发集合通信对计算的干扰 <0.5%(GPU 上 NCCL 与 SM 抢资源通常 >20%)
软件协同HCCL 把集合通信编译成自主执行的 work-queue 子图,无需 host CPU 介入(SC'26 论文)
性能150B 参数推荐模型(40 加速器)上通信比等价 GPU 集群 快 3.9×;单 rack 带宽最高 940 GB/s
配套OCP 2026 公开的 Tahan 51.2T 以太网 scale-up 交换机(4 台组 16 加速器无阻塞域),与 scale-out 共用同一 RoCE 传输栈

MTIA 300 验证了 Meta 的"PyTorch Native + 通信卸载"路线:当集合通信不再占用计算单元,训练有效算力随规模下降的曲线被显著拉平。

5.2 网络模块 ​

组件用途
Arista 7800RoCE 集群的 spine/核心交换机
Wedge 400 / Minipack2OCP 开放的 ToR/leaf 交换机(RoCE 集群)
NVIDIA Quantum-2InfiniBand 集群的交换机
拓扑rail-optimized fat-tree,400 Gbps 端点
MetaRoCE(2026)自研 RDMA 传输层:有损以太网设计、逐路径状态、报文喷洒、256-bit 选择确认;1% 丢包下 ~86% 吞吐;OCP 开源
FPF(OCP 2026)Flattened Planar Fabric:每 GPU 8×100G 接入 8 平面,3 层 Clos → 2 层,NIC 喷洒 + 主机可绕开故障平面
IceCube / IcePack(OCP 2026)102.4T 交换刀片(Tomahawk6C / Spectrum-6 两版,FBOSS + OCP SAI)/ 首个液冷网络机柜(ORV3-HPR)
Tahan(OCP 2026)MTIA 300 的 51.2T 以太网 scale-up 交换机(4 台组 16 加速器无阻塞域)
BAG(OCP 2026)petabit 级 super-spine,跨建筑/园区/区域互联 Prometheus 与 Hyperion(Inter-BAG 数十 Pbps)
OSPA / ESUN(OCP 2026)开放交换机平台规范(Meta/Microsoft/Alibaba 共导)/ scale-up 以太网数据面倡议(9 家厂商)

网络模块的关键不是单台交换机,而是拓扑 + 调度 + 传输层 + NCCL 协同(见 第 3 章 3.3 节)。2026 年之后,Meta 把每一层都做成了开放规范。

5.3 存储模块 ​

组件角色
TectonicDC 级分布式 FS,Flash 优化,EB 级;承载训练数据与 checkpoint
FUSE API自研 Linux FUSE 接口,跑在 Tectonic 上,给训练提供文件语义
Hammerspace并行 NFS,交互式调试与数据探索
YV3 Sierra Point + E1.S SSD高密度存储服务器 + EDSFF SSD

存储模块的 SLA 核心:checkpoint 跨数千 rank 数百毫秒存取(依赖 Tectonic Flash 吞吐)。

2026:AI 存储蓝图重构(BLOB 化 + 分层缓存) ​

2026-07-01《Meta's AI Storage Blueprint at Scale》(engineering.fb.com)披露了存储栈重构(详见 第 3 章 3.4 节):

  • Tectonic 升级为区域级多租户块存储 fabric,数百个 EB 级集群;训练访问从 NFS 式接口迁移到与 GPU 区域共置的 BLOB API(ZippyDB 扁平元数据 O(1) 解析,fat-client SDK 直读)。
  • 分层缓存:L1/L2(GPU 宿主机内存 + 本地 Flash)→ L3(区域 Flash BLOB)→ 全球 HDD BLOB 真相源;分布式缓存复用 Owl 组件,平均命中率 ~80%。
  • 平滑尖峰:hedged reads 对冲慢节点 + 客户端动态并发控制平滑 checkpoint 写出;prefetch() API 消除跨区导入作业。

5.4 集群管理与调度模块 ​

组件角色
Twine集群管理器(cluster manager),把数据中心抽象成资源池;从推荐系统时代延续
MAST训练/通用调度器;支撑 Prometheus 的长距离地理分布式训练

⚠️ Meta 公开资料无"Titanium 调度器";集群管理用 Twine,训练调度用 MAST。

5.5 可靠性模块:SDC 三件套 ​

这是 Meta 在 GenAI 时代最具辨识度的工程。硬件故障分三类:

类别特征治理
Static开机即坏的二元故障(不枚举/不上电)健康检查 + 维修,规模大但易 triage
Transient热失控、不可纠正错误的间歇故障用人造负载在非生产期复现 + "sticky" 状态遥测
Silent(SDC)硬件算错但不留痕迹,硅缺陷导致;约每 1000 设备 1 个故障(软错误的 1000×)三层检测(见下)

SDC 三层检测 ​

机制工作方式覆盖周期局限
Fleetscanner维护期(固件升级/维修)跑定向微基准,把签名纳入遥测45–60 天扫一遍全集群专用测试 host,对部分 SDC 太慢
Ripple与负载同机共存,跑 ms–s 级测试,跨核/线程重叠指令几天扫一遍全集群比 Fleetscanner 快
Hardware Sentinel内核态,评估应用异常,test-agnostic持续(分析面)比 testing-based 方法高 41% 覆盖率

训练 SDC 的两种失败模式 ​

  1. NaN 传播:SDC 把一个合法值推成 NaN → 感染后续计算 → 扩散到整个 accelerator domain → host → 整个集群停顿。源往往只是单个加速器上的几次特定计算,但在集群规模下难以定位。
  2. 梯度方差污染(corrupted gradient variance):SDC 影响梯度计算(爆炸/塌缩/陷局部最优),但数值在界内、被当作正确,在同步训练里作为"真值"交换 → 训练看似在前进实则没进步。比 NaN 更难检测,可能数周/数月才显现。

SDC 治理手段 ​

基础设施策略(infrastructure,集群级 triage):

  • Reductive triage:二分搜索 + mini-training 迭代,逐步缩小集群隔离 NaN 源;用新节点重组集群从 checkpoint 恢复。
  • Deterministic training:跑已知有效模型几步,验证非数据相关的计算故障。
  • Hyper-checkpointing:提高 checkpoint 频率,把 NaN 困在特定 accelerator/host,加速 triage。

栈策略(stack,与 workload 协同):

  • Gradient clipping:限制梯度范围,把超界值裁剪、检测 NaN。
  • Algorithmic fault tolerance:把容错融进训练算法,降低对检测/triage 的依赖(CPU 训练已验证)。
  • Tri-variate computational training:用 shadow node 在随机迭代重复训练 step,shadow 与 live 不一致则只调查那批节点。
  • Parameter vulnerability factors(PVF):识别脆弱层 vs 韧性层,把脆弱层映射到韧性硬件。
  • Divergence detection:为每个神经元维护分布图,检测推理期腐败(采样率可控)。

工厂到机队(Factory-to-Fleet) ​

Meta 把 SDC 治理延伸到硅的全生命周期:设计期(RAS 架构、telemetry、debug hook)→ 验证/集成期(yield 分析、制造诊断、机队签名反馈)→ 部署期(用户态诊断、周期测试、覆盖图)→ 运营期(Hardware Sentinel 大规模分析、固件 hook)。叠加栈级韧性(firmware/compiler/kernel/OS),把可靠性做成"贯穿硅生命周期 + 软硬协同"的体系工程。

可观测性与自动化 RCA:Zoomer / DrP(2025 末) ​

可靠性工程的另一半是"出问题后多快定位"。Meta 在 2025 年底公开了两个平台(详见 第 4 章 4.1.6 节与 第 8 章):

平台时间定位关键数字
Zoomer2025-11-21AI 性能调试与优化平台(straggler 检测、关键路径分析、一键 QPS 优化)100k+ GPU 工作负载;广告训练时间 -75%、功耗 -78%;32k GPU 一天 +30%
DrP2025-12-19根因分析平台:把人工 playbook 排查变为自动执行的 "analyzer" 工作流300+ 团队、~2,000 个 analyzer、每天 5 万次自动分析;MTTR 降低 20–80%;arXiv:2512.04250

两者共同指向 Meta 的 AI4Ops 方向:让 AI 系统自己诊断、自己优化。

5.6 服务模块:Llama Stack 与 Muse 服务栈 ​

Llama Stack 把 Llama 的服务标准化为 9 类 API:

API作用
Inference统一推理接口(跨硬件/框架)
SafetyLlama Guard / Prompt Guard 集成
Agentictool-calling / memory / shielding
Memory会话/长期记忆
Eval标准化评测
Telemetry可观测性
Post-training微调
Toolchain构建/打包
RAG检索增强

2026 年起 Meta 的开发者服务栈转向 Muse 系列 + Meta Model API(dev.meta.ai 模型中心):闭源旗舰 Muse Wonder(API 与订阅提供)与开放权重 Muse Spark Open(SGLang / vLLM 双引擎 + Helion kernel,见 第 4 章 4.3 节)并行。Llama Stack 的"标准化接口"思想被继承——只是落地的载体从单一 Llama 生态扩展为多引擎兼容矩阵。

Llama Stack 与 LLM Gateway、Agent Runtime 的理念相通:用开放协议把"换底层实现不影响上层应用"标准化。

5.7 开放安全工具 ​

Meta 把安全工具全部开源,与 安全篇 互补:

工具作用
Llama Guard 2输入/输出安全分类器,基于 MLCommons 风险分类法
Prompt Guard检测 jailbreak 与 prompt injection
CyberSecEval 2网络安全风险评估
Code Shield代码安全(防不安全代码生成)
GOATGenerative Offensive Agent Testing,模拟多轮中技能对抗者,自动化红队

小结 ​

Meta 的核心模块围绕"大规模同步训练 + 开放发布"组织:计算用 Grand Teton / Catalina / MTIA / MI300 异构组合;网络用 RoCE + InfiniBand 双织物 + fat-tree;存储用 Tectonic + Hammerspace;调度用 Twine + MAST;可靠性用 SDC 三件套 + factory-to-fleet;服务用 Llama Stack;安全用 Llama Guard/Prompt Guard/CyberSecEval。每个模块都既是"工程产物",又是"开放标准"——这正是 Meta 基础设施的双重身份。

Released under CC-BY-SA-4.0 License.