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
• Building Documentation
  • Building the API Documentation
  • General Documentation Guidelines
  • Adding Documentation to Python Code
  • Adding Documentation to C++ Code
  • Sphinx Documentation Pointers
• Contributing to FBGEMM / FBGEMM_GPU
  • Code of Conduct
  • Pull Requests
  • Contributor License Agreement (“CLA”)
  • Issues
  • License
• Contact Us
  • GitHub
  • Slack

FBGEMM_GPU Overview

• Jagged Tensor Operators
  • High Level Overview
  • Jagged Tensor Format
  • Jagged Tensor Operations

FBGEMM_GPU Python API

• Table Batched Embedding (TBE) Operators
  • SplitTableBatchedEmbeddingBagsCodegen()
• Jagged Tensor Operators
  • jagged_2d_to_dense()
  • jagged_1d_to_dense()
  • dense_to_jagged()
  • jagged_to_padded_dense()
  • jagged_dense_elementwise_add()
  • jagged_dense_elementwise_add_jagged_output()
  • jagged_dense_dense_elementwise_add_jagged_output()
  • jagged_dense_elementwise_mul()
  • batched_dense_vec_jagged_2d_mul()
  • stacked_jagged_1d_to_dense()
  • stacked_jagged_2d_to_dense()

FBGEMM_GPU C++ API

• Sparse Data Operators
  • CUDA Operators
  • 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