FBGEMM 和 FBGEMM_GPU 文档主页¶
欢迎来到 **FBGEMM** 和 **FBGEMM_GPU** 库的文档页面!
**FBGEMM** (Facebook 通用矩阵乘法) 是一个用于服务器端推理的低精度、高性能矩阵-矩阵乘法和卷积库。此库用作 x86 机器上 PyTorch 量化运算符的后端。
**FBGEMM_GPU** (FBGEMM GPU 内核库) 是一个用于训练和推理的高性能 PyTorch GPU 运算符库的集合。此库构建在 FBGEMM 之上,并提供高效的表批量嵌入包、数据布局转换和量化支持。
目录
FBGEMM 开发
FBGEMM_GPU 概述
FBGEMM 稳定 API
FBGEMM C++ API
FBGEMM_GPU C++ API
FBGEMM_GPU Python 模块 API