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 and 11.8 in CI, and with PyTorch packages (2.1+) that are built against those CUDA versions.
See the full Documentation for more information on building, installing, and developing with FBGEMM_GPU, as well as the most up-to-date support matrix for this library.
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.8.0rc0-cp312-cp312-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 2610ae5d88dc7caaa601db1f517d914dbd74b7d31b54511e70abafe9a1f8cbc7 |
|
MD5 | 7b88afe27a8ca9ffeb4f1372d0b07cae |
|
BLAKE2b-256 | 926a8fc7e3867e25c700a2e5df082ba63737e51367422918f326aed238df50ec |
Hashes for fbgemm_gpu_cpu-0.8.0rc0-cp312-cp312-manylinux2014_aarch64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | cb3563eab2eb5fcf2d4122f9bab545faa81ec64979a493554ab21ae54ae220bf |
|
MD5 | 7329cf5c97b87005f3290445e590ae9c |
|
BLAKE2b-256 | f3325fe6b953990a286d4372c36ca6d8eb222f637015ef2b1c47aa0a0f35cb17 |
Hashes for fbgemm_gpu_cpu-0.8.0rc0-cp311-cp311-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 1226a67fa3038932602eb518476c948036402264754fad9fe69a41e7e32f690e |
|
MD5 | 92916a9eae567b5fe7d2886519ae3ff7 |
|
BLAKE2b-256 | 6e3ea7834ed990075715ba518eb2f13600e9612b0241fa836e8c2f86f24debc8 |
Hashes for fbgemm_gpu_cpu-0.8.0rc0-cp311-cp311-manylinux2014_aarch64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | f359023d5ff6bb54cc75c0a10b92b1da1f6ab0fdb6c65fb33c30473b1dbb2887 |
|
MD5 | 4309aebaeeeae1f6bec4e8c9d3bef71c |
|
BLAKE2b-256 | c081b69bc48e082b8bc2e67c5b5ede915bf6b74f597ea26777629c7b91ddfa17 |
Hashes for fbgemm_gpu_cpu-0.8.0rc0-cp310-cp310-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 570b760d9a3c5d4b5ccbfb6ce9c6c93ad796895edb7424fa82e661ea5b31954b |
|
MD5 | 516b258df38444a0f52e23b9e9f180cc |
|
BLAKE2b-256 | d41b183dffe96b299ec964ce18621d61354e49218854ebf09def1404d756ee31 |
Hashes for fbgemm_gpu_cpu-0.8.0rc0-cp310-cp310-manylinux2014_aarch64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | f843d635979c95d83a1471c49dd3d0b8bda1ddf2ecc8af491977d787916237ae |
|
MD5 | 6927e3b12a0835c5ab9b6d0ba238cf24 |
|
BLAKE2b-256 | c0f3a1f16b1c822eb61cef031851ac4a057a389a26460096e96db42553410bbf |
Hashes for fbgemm_gpu_cpu-0.8.0rc0-cp39-cp39-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 3d6affd80f2ef9b0b5fa890427a36d3e1dd78ba2b2d7bc1202402bd11a0a0f99 |
|
MD5 | 7572e30d8fa2d16ab02611ba6891b64f |
|
BLAKE2b-256 | 084a4fb5c95c74d98b93b99503d77fd5cacab02df256d393a8939d7e407e4a4e |
Hashes for fbgemm_gpu_cpu-0.8.0rc0-cp39-cp39-manylinux2014_aarch64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | d254fed355659a34fd70d4f0d737bed54c5fcc7779e8c042c552b9c182b7be74 |
|
MD5 | 551648f0a85cb123840cc49bbeaaba8a |
|
BLAKE2b-256 | bcf81e1247b2a4c24b57edf92ab2a69c1ff5e9c65ac9a92df8bf08308021ba89 |
Hashes for fbgemm_gpu_cpu-0.8.0rc0-cp38-cp38-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 435c355eb4b804ed5e4e2babc02e33f4687de6a35ab6d914fd5663e9668d4c94 |
|
MD5 | 2a3ea5a369880f09c0f883ee2e976943 |
|
BLAKE2b-256 | 44d31e045ffd4a10fab51fce05b79ce22c3a8decf41a06831a1be7d20dff7ab2 |
Hashes for fbgemm_gpu_cpu-0.8.0rc0-cp38-cp38-manylinux2014_aarch64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | bba80edba495f27b9fa1cdf8cf0d6ef603850724be7665392003ed7baac99061 |
|
MD5 | 8568f52a6be8e340d65dde48a087b52a |
|
BLAKE2b-256 | 21b419a6babd7c64e36b9b0963e1593488a51747b6eba0aae3ae8414dc9202f4 |