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.0rc4-cp312-cp312-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 34788ea9f80a981a2ebab1e5507783c931260f1d53dcf428e00269f5d400849f |
|
MD5 | 2b79b194e6bc72a5ce9fd768036e3cb3 |
|
BLAKE2b-256 | 569d67721193fd16b6473bb63a78b8eddb98c9560f9322aa22bb6e08abe62c55 |
Hashes for fbgemm_gpu_cpu-0.8.0rc4-cp312-cp312-manylinux2014_aarch64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 9320e1d45830cf2d8616ac25581efa2defb79b65c10bb914dd0025fc48e703db |
|
MD5 | 0fdf4ef862576ecf44735e87d8eae730 |
|
BLAKE2b-256 | 0c367c84644653889c1906381b4068cb63e629bbcb80b76d6f8218d13e9af0b7 |
Hashes for fbgemm_gpu_cpu-0.8.0rc4-cp311-cp311-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | e1b69cd8b2eba9e59db3ea1674828042510b666616855b4ac21b0c3aed3bb3a2 |
|
MD5 | 849c1357ed5b7feaa140ed553293410c |
|
BLAKE2b-256 | 3540a3661aa8a53e15f18caa7dd01bbd1ca295c072c1df2123a9d874924f4cea |
Hashes for fbgemm_gpu_cpu-0.8.0rc4-cp311-cp311-manylinux2014_aarch64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | d1758661ccca05e665da981983f1ce8a05374739c08370aaadf4ec45f8b025f2 |
|
MD5 | db757ef52c97dde0b40752627af67304 |
|
BLAKE2b-256 | e44ce291337ceda043e8b20683e6318f7452faf659bc9b9a427c351c9655c210 |
Hashes for fbgemm_gpu_cpu-0.8.0rc4-cp310-cp310-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | cb340a94da2a581cfb6b52aedffa71c2f120b0fed0f6ab7dffaeb1b3c780393e |
|
MD5 | 9c09b37b3b84fbd9fd2c56d0d9858504 |
|
BLAKE2b-256 | 3881b1f52035ff5225964c2aaf339b452c637f12034af967a7f2dbfaef2df1ce |
Hashes for fbgemm_gpu_cpu-0.8.0rc4-cp310-cp310-manylinux2014_aarch64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 129fe05f9178f33ab5573014bb3ed595fb75196a25051ea2478e0d5e48916d86 |
|
MD5 | 77d3fd479c6f98716f424835d4569c3e |
|
BLAKE2b-256 | e271c30a929257997b4ad65d6736b87e0eec005ac8d766dd7bf67ae601c7febe |
Hashes for fbgemm_gpu_cpu-0.8.0rc4-cp39-cp39-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 2ce8e615f9e87e03edc57336f479081e51ec7081c71fb50e39969d3220047182 |
|
MD5 | 76ac7187abce091d5a902a3ed46c6734 |
|
BLAKE2b-256 | f67c96087991073f9e754941fababdb1170b65d980d8719e6ca230f48e907079 |
Hashes for fbgemm_gpu_cpu-0.8.0rc4-cp39-cp39-manylinux2014_aarch64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 4cad547a1eb845ab210925a0d9656214d216aa02027402f76960bc0e912a8a0b |
|
MD5 | 2fef8136c7c1affdca098bf4b4233923 |
|
BLAKE2b-256 | 9e52cfcfc0289993caa4ed83a142645bc1e13e58b32f44fb431150d57cb41714 |
Hashes for fbgemm_gpu_cpu-0.8.0rc4-cp38-cp38-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | c16712353b3db8092b7c886590cc80081fe3a30eee608a540c09787ca8b7e3e2 |
|
MD5 | 4c7e8a6035c703debd930d5f02874887 |
|
BLAKE2b-256 | e447142982b9928d45144dff77623c008f9ffbaca30538dcaeeebb15030cd0c7 |
Hashes for fbgemm_gpu_cpu-0.8.0rc4-cp38-cp38-manylinux2014_aarch64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | daed682f432626ad7f7fa8d69a5961be0dd9426f24a0b0cb3040d3da9f66903c |
|
MD5 | 1026cefeac4044e5ce4e3c88cfae40a0 |
|
BLAKE2b-256 | 5a3fd9e2b4346422921e55b165d6d3d9239d0abbdb1e02acdaa6c6f67ea48e63 |