跳到正文
热点事件持续更新

Meta 将推荐系统 TBE kernel 从 CUDA 迁移到 Triton

1 篇报道1 个报道来源3 小时前更新

先了解这件事

AI 综述

Meta 团队将推荐系统核心的 Table Batched Embedding(TBE)kernel 从 CUDA 迁移到 Triton,这一 Triton 实现被称为 FBTriton。据 2026 年 10 月 7 日 AGI Hunt 的报道,迁移后前向传播最高提速 1.28 倍,反向传播最高提速 2 倍;该报道将其概括为 PyTorch 用 Triton 替换 CUDA。目前公开的信息仅涉及这次迁移本身及上述性能数字,尚无关于落地范围、覆盖模型规模或后续计划的更多披露。

AI 根据报道生成 · 2 小时前更新

报道时间线

沿着报道,了解事件的不同侧面。

10月7日
  1. AGI Hunt
    PyTorch 用 Triton 替换 CUDA,推荐系统 kernel 前向提速 1.28 倍

    Meta 团队将推荐系统核心的 Table Batched Embedding(TBE)kernel 从 CUDA 迁移到 Triton(FBTriton),前向传播最高提速 1.28x,反向传播最高 2x。

本事件热度走势

还没有足够的连续观测数据,暂不绘制趋势。