乐于分享
好东西不私藏

Meta300自研AI芯片,主打“带宽比算力重要的“通信原生”AI芯片

Meta300自研AI芯片,主打“带宽比算力重要的“通信原生”AI芯片
Meta 认为 带宽比算力重要!“通信原生”更重要
Meta最新披露的MTIA 300便是其中的佼佼者。
作为Meta首款将后端网络直接集成到芯片封装内的加速器,MTIA 300不仅在硬件规格上实现了飞跃,更在架构层面解决了大规模分布式训练中“计算与通信难以兼顾”的行业痛点。

为什么需要MTIA 300

传统的GPU在处理推荐系统(DLRM)训练时面临一个尴尬的困境:
DLRM包含大量的Embedding查表和集合通信操作(如AllToAll),这些操作频繁且数据量大。如果让负责计算的SM(流多处理器)去处理通信,就会挤占宝贵的计算资源;如果不处理,通信延迟又会拖累整体性能。
MTIA 300的设计哲学是“通信原生”。
它是业界首款内置NIC Chiplet并配备通用集合卸载引擎的加速器,旨在彻底解耦计算与通信,让两者能够真正并行执行。虽然最初为DLRM设计,但其架构优势已成功扩展至后续的GenAI训练与推理场景。

:Chiplet设计与三大创新

MTIA 300采用了先进的Chiplet(芯粒)架构,
由1个计算芯粒和2个网络芯粒组成,通过3.2倍面积的Interposer互连。

1. 规格对比:相比前代MTIA-2i的全面进化

 0cm 0cm 0cm 0cm;">

特性

MTIA 300

MTIA-2i (前代)

提升/变化

频率

1.9 GHz

1.35 GHz

+40%

规模

72亿晶体管, 5.11亿 FLOPS

23.5亿晶体管, 1.03亿 FLOPS

~3x 算力密度

TDP功耗

912W (典型667W)

85W (典型65W)

面向数据中心级高性能

主机接口

PCIe Gen5 x16 (64 GB/s)

PCIe Gen5 x8 (32 GB/s)

带宽翻倍

Scale-up网络

16节点域, 800-1000 GB/s

N/A

新增片间高速互连

Scale-out网络

4096 L1节点, 200 GB/s

N/A

支持超大规模集群

封装尺寸

77.5mm × 77.5mm

50mm × 40mm

更大封装容纳更多IO

MTIA 300不再是单卡作战,而是原生支持16节点的Scale-up域和无限扩展的Scale-out网络,这是支撑万卡集群训练的物理基础。

2. 计算单元(PE):双核RISC-V + 冗余设计

每个处理单元(PE)内部包含:
双RISC-V向量核心:提供对称编程模型,64B宽向量指令集,提升SIMD吞吐。
512KB本地存储(LS):软件管理的循环缓冲区,减少对外部HBM的访问压力。
内存桥接器(MB):连接所有组件的内部NoC,集成中断控制器与调试模块。
硬件冗余:由于采用光罩极限尺寸设计,计算芯粒包含一行冗余PE。若某列PE故障,可在启动时透明替换,不影响NoC性能,显著提升良率。

3. 消息引擎(ME):通信卸载的灵魂

这是MTIA 300最核心的差异化特性。ME配备了近存计算(Near-Memory Compute, NMC)单元,能够独立于主计算网格处理集合通信操作。
完全卸载:AllReduce、AllGather、AllToAll等操作由ME自主执行,不占用PE资源。
RDMA原生支持:内置NIC Chiplet直接处理网络协议栈,无需外部网卡中转。

计算与通信的完美重叠

MTIA 300的实际表现如何?

1. 集合通信性能 vs H100

在16/32/64 ranks的规模下,MTIA 300在多种消息尺寸的集合操作中展现出强劲竞争力:
大消息优势:得益于更大的Scale-up域和2.2倍的Scale-up带宽,当消息大小超过16MB时,MTIA 300显著优于H100+NCCL组合。
小消息优化空间:目前小消息性能略逊于H100,但这是因为软件栈尚未完全针对小消息优化,且在实际DLRM训练中,小消息仅占极少的墙钟时间。

2. 计算-通信重叠效率(Overlapping Efficiency)

这是衡量AI芯片“实战能力”的黄金指标。测试显示,在同时运行4K×4K×4K TF32 GEMM和集合通信时:
MTIA 300:计算效率和通信效率均维持在接近100%,几乎零干扰。
H100:出现明显的SM资源争抢,导致并发执行时性能显著下降。
这种“双线满跑”的能力,正是MTIA 300针对DLRM高频通信特性量身定制的成果。对于依赖大量Embedding交互的推荐模型而言,这意味着实打实的训练加速。

HCCL编译式通信库

硬件的强大需要软件的释放。Meta为MTIA 300开发了专用的集合通信库HCCL,其设计理念与传统GPU通信库截然不同。

编译式通信模型

HCCL不在设备上动态决策通信算法,而是由Host端生成完整的通信子图(Subgraph),包括依赖关系。ME自主执行这些子图,实现:
设备触发(Device-Triggered):计算内核通过信号量直接唤醒ME执行通信,避免CPU调度开销和Graph Break。
预发布接收WQE:ME可提前准备好接收描述符,减少“Receiver not Ready”导致的NAK重传。
拓扑感知算法选择:自动识别非对称的Scale-up/Scale-out网络拓扑,选择最优通信路径。

实际效果

训练场景:机架内集合通信带宽高达940 GB/s
推理场景:通过单边通信和设备触发机制,实现极低延迟响应。
MTIA 300不仅仅是一颗芯片,它代表了Meta对AI基础设施的深度思考:当通用GPU无法满足特定工作负载的极致效率需求时,软硬协同的专用架构才是破局之道。
首创性:首款集成NIC Chiplet + 通用集合卸载引擎的AI加速器。
实用性:已在Meta生产环境的DLRM训练中验证,计算通信重叠效率远超竞品。
延续性:其设计原则已被后续MTIA 400/450/500继承,分别对标GB300及未来GPU,覆盖GenAI训练与推理全场景。
随着AI模型规模的持续膨胀,像MTIA 300这样“为通信而生”的架构,或许比单纯堆砌算力更能代表下一代AI芯片的发展方向。
全国首个算力和大模型工程专属服务IP
“算力百科”是全国首个算力和大模型工程专属服务 IP品牌,专注于国内AI算力行研;团队来自顶级芯片设计、算力中心构建及大模型开发一线,贯通“芯·算·模”,从实战角度出发,构建国产AI算力新范式,深耕算力和大模型行业!