Welcome to FBGEMM’s documentation!

This documentation provides a comprehensive reference of the fbgemm_gpu library.

FBGEMM_GPU General Info

• Build Instructions
  • Set Up an Isolated Build Environment
  • Set Up for CPU-Only Build
  • Set Up for CUDA Build
  • Set Up for ROCm Build
  • Install the Build Tools
  • Install PyTorch
  • Build the FBGEMM_GPU Package
• Installation Instructions
  • Set Up CPU-Only Environment
  • Set Up CUDA Environment
  • Set Up ROCm Environment
  • Install Python Libraries
  • Install PyTorch
  • Install the FBGEMM_GPU Package
  • Post-Installation Checks
• Testing FBGEMM_GPU
  • FBGEMM_GPU Tests
  • FBGEMM_GPU Benchmarks

FBGEMM_GPU Python API

• Table Batched Embedding (TBE) Operators
• Jagged Tensor Operators

FBGEMM_GPU C++ API

• Sparse Data Operators
  • Sparse Data CUDA Operators
  • Sparse Data CPU Operators
• Quantization Operators
  • CUDA Operators
  • CPU Operators
• Pooled Embeddings Operators
  • Merge Operators
  • Permutation Operators
• Table Batched Embedding Operators
• Jagged Tensor Operators
  • CUDA Operators
  • CPU Operators
• CUDA Memory Operators
• Combine Input Operators
• Layout Transformation Operators
  • CUDA Operators
  • CPU Operators
• Embedding Operators
  • CUDA Operators
  • CPU Operators