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 12.1.0 and 11.8 in CI, and with PyTorch packages (2.1+) 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_cpu-0.6.0rc2-cp312-cp312-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | b21e7d6c9e79acec0026d5b6fcfbdc3b1f2c41f36febd5530299ee4b884a94cc |
|
MD5 | 0a9ace8455d2bd7b038d36e954667f94 |
|
BLAKE2b-256 | e69fdda438eca4e5d3a115be3327b6ffc94e4f7e5c1965e30d805d5457e9f9ef |
Hashes for fbgemm_gpu_cpu-0.6.0rc2-cp312-cp312-manylinux2014_aarch64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | fa7f836c18f6ca882d6ad4b29f80df6c769b68e57924bb7fd125ac0005c9a926 |
|
MD5 | 18a4b6d59ba1a2725092f75374eb6dde |
|
BLAKE2b-256 | 0103b6d46cb0160ebd95794389aff88c2b68b168f583d35c1e65f63d84acb433 |
Hashes for fbgemm_gpu_cpu-0.6.0rc2-cp311-cp311-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 5d3d62beee8683f4c0138a0b1d0d4bde4152212ec0cb530cd6f0985010e338bc |
|
MD5 | 1e4952d8fd350b77439224e99b3cbb87 |
|
BLAKE2b-256 | 7accf5735bb2a21fbd361b281215f717113cba29198b97af4e2f58145cf55b48 |
Hashes for fbgemm_gpu_cpu-0.6.0rc2-cp311-cp311-manylinux2014_aarch64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | c5400fbe13687d8d7f1d1e77c333ed1f2fef3a243b62d510a541ee560f97c105 |
|
MD5 | 2fff66f9baa33fd29b7e49d3d28ee0df |
|
BLAKE2b-256 | 6511618b9cee2fdb1a97fd4ddae00255f83eef04437fbd1696e7285b9a0ff47a |
Hashes for fbgemm_gpu_cpu-0.6.0rc2-cp310-cp310-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 3a29858dcc2d8cc3431a74de36e4d2b9f5266f56ad3543d9306bcc287245d37a |
|
MD5 | 2aa71842fc42796713f321fb9e3fda23 |
|
BLAKE2b-256 | 4de03392ab9e21b93bfe832c80ac87d36cb19f4d49dd042fd6c34bd5a5ffbda6 |
Hashes for fbgemm_gpu_cpu-0.6.0rc2-cp310-cp310-manylinux2014_aarch64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | c8c2ec14257530a7c9743a0456196f12ec3ac40b02f12f776dcfd252f09da017 |
|
MD5 | 863fd01988871622ae1f3bdbee254248 |
|
BLAKE2b-256 | f49a1a7c7868cf08575f0748283f7a9a2dcb602a15ce2c1ed1b5da0649c000e9 |
Hashes for fbgemm_gpu_cpu-0.6.0rc2-cp39-cp39-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | bcd028260633103b9c76b7484feda625ce905841643b8e3fd204bb89d4dc8639 |
|
MD5 | 22d4da29b35a5151c4d1705001ea87b4 |
|
BLAKE2b-256 | 86353669a1008e619370e0ffefb2bff7cd31c8af687b2fcc028ccf4d813aede6 |
Hashes for fbgemm_gpu_cpu-0.6.0rc2-cp39-cp39-manylinux2014_aarch64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 043cb6ec3dd0ed8512bd7de198f72eb24d8aa6502f416a1b646acf9d36e4c230 |
|
MD5 | 06692bcf19763bd77a57673e34c85a65 |
|
BLAKE2b-256 | 2a6c1976f55f5881aafe66bed07f139a932bfd9ed89e1960a0fb92dfbfae026f |
Hashes for fbgemm_gpu_cpu-0.6.0rc2-cp38-cp38-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 38c69e1ca201ab97636473496948ed89b0264ea22d5f5ef228bad8724b64df61 |
|
MD5 | b4f1ea95dbc237041d819c97d0efd5c2 |
|
BLAKE2b-256 | ebb3076e1eb5f5892d1d28a2fcdf03e4a98ec494ea090c5537410e6b8d192fec |
Hashes for fbgemm_gpu_cpu-0.6.0rc2-cp38-cp38-manylinux2014_aarch64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 4dcfea248d87555f067fce6ad4413058c9b2fe5b84052165cc333ef597d12448 |
|
MD5 | f855f8b368a3a3177a98a198ab35f109 |
|
BLAKE2b-256 | 03d74ce3988f3a77786f7f2499ecadbfb2ea43ba52d3b25e150423dd40aff5f3 |