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_nightly_cpu-2024.5.18-cp312-cp312-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | d8c15227748304631d11babc1ba8637ded44312dbf881d3ed135bd512d60c9c2 |
|
MD5 | dec030ff465b3bdec65bf17814f83c28 |
|
BLAKE2b-256 | de5efeadf7d1a330e519310e218a47d4b9ba7c6a59fe4ff43dcebe509ff6f83e |
Hashes for fbgemm_gpu_nightly_cpu-2024.5.18-cp312-cp312-manylinux2014_aarch64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | a03ea901efeabc710a777aea6639a4922a83a268f536edc4c5f2192daa6c0ca6 |
|
MD5 | 5cef883eda3b2e551888367536eb3cd1 |
|
BLAKE2b-256 | 3d520752d32e84628029eccb0eb2563b5c1c135c20396b6bac664f7ec63c68aa |
Hashes for fbgemm_gpu_nightly_cpu-2024.5.18-cp311-cp311-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 599f0bffd6449e75aece625b856a00fa8539e03e75ad84d7e81106d98068d735 |
|
MD5 | 61dcc129e7ec0a7d60f7909a40b37b23 |
|
BLAKE2b-256 | 452829417a8a037fdc80e805a976905bf6da981732de9ee82baa2e5ff58ebdaf |
Hashes for fbgemm_gpu_nightly_cpu-2024.5.18-cp311-cp311-manylinux2014_aarch64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | da490724ef46ded9669b9f2a43f2e2e78e984da602942c8fc6a23d632fd26b02 |
|
MD5 | b12ef6102234bef94dbbb13a43c437fc |
|
BLAKE2b-256 | a8b68560820f9b4ac02ced2816c9287a1791b6e753cf045595ab3ead2bf043e2 |
Hashes for fbgemm_gpu_nightly_cpu-2024.5.18-cp310-cp310-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 294b357ff7bfdde7dde8fde74968834299c3a128554022b47d7f98d7e2c85106 |
|
MD5 | eb0a11acc81b85fd10b0f1215eda3755 |
|
BLAKE2b-256 | f3439f6151358d240f0feb10cbe85072eec996deaf95f6cb068e64f2ccb88720 |
Hashes for fbgemm_gpu_nightly_cpu-2024.5.18-cp310-cp310-manylinux2014_aarch64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | fd0bd08812995424dc83b1f6c7f80726eb3c7d487c775f754acf9deaa7623a29 |
|
MD5 | 3c3cb2abdf998e229be2cacd9b5fa067 |
|
BLAKE2b-256 | 0065e4018dfa87edb4384b99a40d07f87bdac4950fd3df35a04dad346b83b2e1 |
Hashes for fbgemm_gpu_nightly_cpu-2024.5.18-cp39-cp39-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 31e1047509693cedac1b4432ae2a6206b8c7fedc622ead0633d6da65dbd02743 |
|
MD5 | 60e62867ed4164afe9476a55b76f34a7 |
|
BLAKE2b-256 | 8d0300d5825e94b60476ed28079137936e5477c289baf7bce294913cc644c474 |
Hashes for fbgemm_gpu_nightly_cpu-2024.5.18-cp39-cp39-manylinux2014_aarch64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | a7536cef51b8c16162a44bba792ba883301b566cca1b16b023799baf7684374e |
|
MD5 | 9fd415407b53d143ae2138f83b31394e |
|
BLAKE2b-256 | 30ac8b28d3f04c1bfd46849c82f2d5f21129921efd60ccc753dc68e7776ec2ba |
Hashes for fbgemm_gpu_nightly_cpu-2024.5.18-cp38-cp38-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 6a0cafaff5d62d1bde995b57e1b356cdb87aedb14612cee422e798f35f178640 |
|
MD5 | 06e60a29b0c479f5d06cd8e1d4a3c38d |
|
BLAKE2b-256 | 9e218fda54a52ec23bea1eba97d1db9a5fcb3372c08fa33b3f63de42c952a3cf |
Hashes for fbgemm_gpu_nightly_cpu-2024.5.18-cp38-cp38-manylinux2014_aarch64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 215f4d6e69b1698228687b6630564101a173c102e1dbd1cdb72eef9d3eec10b5 |
|
MD5 | c060a8f98ddb79b95886848fe3f0f789 |
|
BLAKE2b-256 | 1069ca0a9671311ebf859aebd87782ec50a666b2860d876bffce9f43817b9f0c |