开发与工具

使用 FBTriton 改进表批量嵌入运算

Modernizing Table Batched Embeddings with FBTriton

PyTorch Blog发布于 10月7日 1 分钟阅读

专题配图图片仅供信息参考图片仅供参考版权归原作者

摘要

本文探讨用于表批量嵌入（TBE）前向和反向传播的 FBTriton 内核设计。这些核心算子处理推荐系统中分布在数千个分片 GPU 上的嵌入查找……本文最初发布于 PyTorch。

原始资料

PyTorch Blog · 来源摘要 · 正文节选

查看来源原文产品参数、测试条件及最新修订以来源原文为准。