FBGEMM 與 FBGEMM_GPU 文件首頁¶
歡迎來到 **FBGEMM** 和 **FBGEMM_GPU** 函式庫的文件頁面!
**FBGEMM** (Facebook GEneral Matrix Multiplication) 是一個低精度、高效能的矩陣-矩陣乘法和卷積函式庫,用於伺服器端推論。此函式庫作為 x86 機器上 PyTorch 量化運算子的後端。
**FBGEMM_GPU** (FBGEMM GPU Kernels Library) 是一系列高效能 PyTorch GPU 運算子函式庫的集合,用於訓練和推論。此函式庫建構於 FBGEMM 之上,並提供高效的表格批次嵌入袋、資料版面配置轉換和量化支援。
目錄
FBGEMM 開發
FBGEMM_GPU 概觀
FBGEMM 穩定 API
FBGEMM C++ API
FBGEMM_GPU C++ API
FBGEMM_GPU Python 運算子 API
FBGEMM_GPU Python 模組 API