No project description provided
Project description
FBGEMM_GPU
FBGEMM_GPU (FBGEMM GPU Kernels Library) is a collection of high-performance PyTorch GPU operator libraries for training and inference. The library provides efficient table batched embedding bag, data layout transformation, and quantization supports.
FBGEMM_GPU is currently tested with CUDA 11.7.1 and 11.8 in CI, and with PyTorch packages (1.13+) that are built against those CUDA versions.
Only Intel/AMD CPUs with AVX2 extensions are currently supported.
See our Documentation for more information.
Installation
The full installation instructions for the CUDA, ROCm, and CPU-only variants of FBGEMM_GPU can be found here. In addition, instructions for running example tests and benchmarks can be found here.
Build Instructions
This section is intended for FBGEMM_GPU developers only. The full build instructions for the CUDA, ROCm, and CPU-only variants of FBGEMM_GPU can be found here.
Join the FBGEMM_GPU Community
For questions, support, news updates, or feature requests, please feel free to:
- File a ticket in GitHub Issues
- Post a discussion in GitHub Discussions
- Reach out to us on the
#fbgemm
channel in PyTorch Slack
For contributions, please see the CONTRIBUTING
file for
ways to help out.
License
FBGEMM_GPU is BSD licensed, as found in the LICENSE
file.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distributions
Built Distributions
Hashes for fbgemm_gpu-0.5.0-cp311-cp311-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 0e2c85a7e76eaa0b86b0ed1398a74bd7bde0d2d67dec1decb3dd973c4307d443 |
|
MD5 | fd79d1f99edfca4bc6ea08db157af542 |
|
BLAKE2b-256 | baec7f5cb9378324c0179e30dfc02dcea14e36386c5feb2c1ce13467d13583d7 |
Hashes for fbgemm_gpu-0.5.0-cp310-cp310-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 355c758286f13d389ccb2b50a9f9ba2e41c17214ba1f14e862fbefd1192f0fd8 |
|
MD5 | 87d75dbb29101ad66657ba2c829502a4 |
|
BLAKE2b-256 | 56f191227f85f3df61b633ce63b7a86339603551c5fadd47e30db7a3739e2740 |
Hashes for fbgemm_gpu-0.5.0-cp39-cp39-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 653f05796cd866a05b33ba073036dc5a829bbff081295ebab06e09d768caf9aa |
|
MD5 | 217b4c384b02755900b9a591b1ef5eff |
|
BLAKE2b-256 | 35b8b16f068721117c1bdf96651a8caa1ee82c02a56d1c4b6e97b6086058273f |
Hashes for fbgemm_gpu-0.5.0-cp38-cp38-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 2bd1a1bcc3cd1ce9b73dd396bcfc77a6e6aff815d01e9425ec4739185bf8c254 |
|
MD5 | 835fb3dcc9d540ea8f39874fea839d3b |
|
BLAKE2b-256 | 97602d1ae119efe7fb77a6dbad15c29e3849c5c6c55472658f48948b56aebccb |