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