AB
AiBoss站
快讯

Meta 开源 TLX:为 Triton 增加硬件原生 GPU 执行编排层

Meta 工程团队发布 Triton Low-Level Extensions(TLX),在 Triton 编程模型之上加入显式 warp group 与 cluster 级编排,用于 RecSys 训练、推理和大规模 LLM 预训练等生产负载。

核心事实

Meta 工程团队于 2026 年 11 月 10 日在其工程博客发布文章,介绍开源项目 Triton Low-Level Extensions(TLX)。据文章描述,TLX 是 Triton 的扩展,在保留 Triton 编程模型的同时,让开发者显式控制 GPU 执行结构,包括 warp group 与 cluster 级编排、异步数据搬运和共享资源协调。文章称,TLX 已在 Meta 的推荐系统训练、推荐系统推理和大规模 LLM 预训练等生产负载中使用,并覆盖 NVIDIA 与 AMD GPU。

背景与影响

文章指出,Hopper 引入 warp specialization 与 TMA,Blackwell 进一步带来 TMEM、cluster 级执行和 paired-CTA 执行,这些能力更多关乎多个执行单元如何协作,而非单纯的计算指令。CUDA 的 SIMT 模型粒度过细,Triton 的块级模型又把执行结构交给编译器决定,两者之间缺少一层抽象。TLX 将这一层称为 MIMW(Multiple Instructions, Multiple Warps),以 warp group 为基本编排单位。文章还提到,Meta 内部 Triton kernel 数量在约 10 个月内增长约 3.8 倍,同期 CUDA 约 2.0 倍,这一层性能改进可较广泛地传导至训练与推理负载。

限制与来源

上述性能与增长数据均来自 Meta 官方博客自述,未提供可独立复核的基准细节;TLX 的具体功能范围、支持的硬件型号、许可证与可用地区,请以项目官网及代码仓库当前信息为准。本文依据 engineering.fb.com 文章《How We Built TLX: A Hardware-Native GPU Compiler for Production ML at Scale》整理。