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.0-cp312-cp312-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | ced3955267d13702922ddc782609dc9946d1ca854ea32775e0fef9067fafaa46 |
|
MD5 | 1a4aed282bc151b1c993b60a86ac560b |
|
BLAKE2b-256 | 94fd846f29ecf82e506544f7b8e499bdfa2fa07e01af871222be3b976635e5aa |
Hashes for fbgemm_gpu_cpu-0.8.0-cp312-cp312-manylinux2014_aarch64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | cc45c1cfb9d6c7d5147be68e313446f86ba75b2a0d25a1e4799b164165c313d5 |
|
MD5 | 3450a0c62208828f51400fda0ece2cfa |
|
BLAKE2b-256 | 723e18c4d3431f8803b29fd381c0f65fb46145d20335c7f63ce69ea74430ba78 |
Hashes for fbgemm_gpu_cpu-0.8.0-cp311-cp311-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | f40a77bb6479f7fcb6d0dd9147c5633b3ee4680cee6801b04e39af3f44208f89 |
|
MD5 | 19255754a33434284de121eb88263ea4 |
|
BLAKE2b-256 | 0ce8c7c1e1ac6508e6577783816a94c7351dd39688bbe3f4178dcc508292d85e |
Hashes for fbgemm_gpu_cpu-0.8.0-cp311-cp311-manylinux2014_aarch64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | d63cf7109723c2cbc5a9c95916861437b6515b1ba6771a78cc34e2e811d0081e |
|
MD5 | eb52fd64248d35c2b5c59f4f01160035 |
|
BLAKE2b-256 | c8997bd9fd6739c61d04809edb1f63bad81a95f8e043e96c1de139eb1adb1fda |
Hashes for fbgemm_gpu_cpu-0.8.0-cp310-cp310-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 4e96ed85548cf32d799b081cd8d8c40318fdfbbfa8c1bcd081a85dc8adb915ed |
|
MD5 | 9df7aec233113ce994e50956c7695d7a |
|
BLAKE2b-256 | 4637446b18c3fc44b53c68e22625aec24057b2ba353e54e9f5ce9661841e5223 |
Hashes for fbgemm_gpu_cpu-0.8.0-cp310-cp310-manylinux2014_aarch64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | cac45ea21f1d45ad5e30b7fffb75a077376224e11980b6925d9a4b591189c315 |
|
MD5 | 9f18f0f4fdfac8c44e2483887de5a5b4 |
|
BLAKE2b-256 | 57b1c00e0dd41640e052705e3e0bccf0ec156e670f8054b5304c49a5832d7975 |
Hashes for fbgemm_gpu_cpu-0.8.0-cp39-cp39-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 658049c07e32612d62d432c353ea3b4d313918708fdca1f1d8b405100f9d1cb3 |
|
MD5 | 3d2953f111d3b0f0457f67640f49ded0 |
|
BLAKE2b-256 | a4aa9c3ee5ae39448810a68317fbad589676a6bd58d6e814b797d7ba029f6a53 |
Hashes for fbgemm_gpu_cpu-0.8.0-cp39-cp39-manylinux2014_aarch64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 71ca461423b51e69ed43fad8ccd72205f3b52d53bc524160ce689da8fc2ca76b |
|
MD5 | f3614a3206a062f6d4cfe8afa9525e53 |
|
BLAKE2b-256 | ce7f1227748c9c60084fe5827592c38fab26301393460c17a79be848e8b6bfba |
Hashes for fbgemm_gpu_cpu-0.8.0-cp38-cp38-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 99223634fb9e0e210ba94c1dbc4d90c7b7c793d0d28d1340e1c215f8edadb06a |
|
MD5 | 2fad46c2d6b4506e98dd26f034c4ef73 |
|
BLAKE2b-256 | 774d42ab82897bed2bf4897ff78eb285a9882d7e8eb265c78433ffba68f8ebfc |
Hashes for fbgemm_gpu_cpu-0.8.0-cp38-cp38-manylinux2014_aarch64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | c31d3e96fc786359f28b735b5536e02871d6506a66be7e17fbd5acc1a7da1d70 |
|
MD5 | dfaa1a29babfc90ee7d20c23a633b154 |
|
BLAKE2b-256 | f5f68fe21148cfd4a6c3c86cf06be9d98011b7771cff4bc0959e9dde87a31c60 |