Release Notes

Software Requirements

FBGEMM_GPU v0.4.1 has been tested and known to work on the following setups:

• PyTorch: v2.0
• CUDA: v11.7, 11.8
• Python: v3.8, 3.9, 3.10, 3.11

It is recommended to prepare an isolated environment for installing and running FBGEMM_GPU, such as Conda and/or Docker.

Availability

FBGEMM_GPU may be fetched directly from PyPI:

# FBGEMM_GPU (CUDA variant)
pip install fbgemm-gpu==0.4.1

# FBGEMM_GPU (CPU variant)
pip install fbgemm-gpu-cpu==0.4.1

Changes

This is a minor release whose main purpose is to deliver Python 3.11 support.

• [New] Add support for Python 3.11 (#1646)
• [Improvement] Add support for group size > 54 in group_index_select (#1611)
• [Improvement] Implement cache miss emulation in UVM_CACHING (#1637)))
• [Improvement] Add TensorAccessor with memcheck (#1602)