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 11.7.1 and 11.8 in CI, and with PyTorch packages (1.13+) 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.5.0-cp311-cp311-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 57ee5f4d1f6d6791b8aaea5da073b9ecd6230e7a7a419d1c2730a538c270e315 |
|
MD5 | ac25f8865e4271a271580a3c1be2d710 |
|
BLAKE2b-256 | d04e52e69440230d43b10e4e5b307b73d1d278e00cff4593757400d0a146aab3 |
Hashes for fbgemm_gpu_cpu-0.5.0-cp311-cp311-manylinux2014_aarch64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | a18103c88e34d9bbb9bc430720e35fb8a189d03a53065863796737535d570761 |
|
MD5 | 1baa24c75654828ea2b08c44a95d8d5d |
|
BLAKE2b-256 | f20c23495bfd9f4df31da5fc7708d6fd4e71161cbe94b089f56b7170f40fcd17 |
Hashes for fbgemm_gpu_cpu-0.5.0-cp310-cp310-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | e8eeab4031b2b21d35953ce3af5a4101a76f69357308027b5a1bb72158c914e3 |
|
MD5 | f7adf42a7531b9620794cb92363f56c1 |
|
BLAKE2b-256 | 8cf0029c3b8c6c340a24d41141427c5a1d7af9882ec2f46a41a3fa01e28d3d04 |
Hashes for fbgemm_gpu_cpu-0.5.0-cp310-cp310-manylinux2014_aarch64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | d446bac19cdd86792ce2c5ff5ed070bbbe65047f056efa984d7bf7a5724f1c02 |
|
MD5 | 5b8e24e459b6e221606323bc1b0a8178 |
|
BLAKE2b-256 | 218791fdfa1b5a3c107d67e3b3a5a969acee35c9cf1e0a6e4ca104134ce6a4a8 |
Hashes for fbgemm_gpu_cpu-0.5.0-cp39-cp39-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | a18a443646a98db0f11a0ac5bb64fb5bff6c0b91bf71108a69c1d0e2614c8786 |
|
MD5 | 41062eb2947250a9cc12fdd1e1a9d16a |
|
BLAKE2b-256 | 95a60027868caeaf74a4bc3c6187f086d6151448b7210e4317fb3360a3baa2cb |
Hashes for fbgemm_gpu_cpu-0.5.0-cp39-cp39-manylinux2014_aarch64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 7b6f46298d1ce36a2e0a4aff8661d016c865902d9b3843d0114fa1d09c0e8da8 |
|
MD5 | 0adbeca4b6d134e55b1e33fe8c873fa7 |
|
BLAKE2b-256 | 94a32263eb93235fd720441ea5ddb906c14889f8a535112d467164aa868b85b3 |
Hashes for fbgemm_gpu_cpu-0.5.0-cp38-cp38-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | ec5f1aa00536b67e06ce953bf95f75e0fbec902722799bfb06aa7e7bb60bab0f |
|
MD5 | f94d18872b30dc5529df9f93191ec833 |
|
BLAKE2b-256 | 2308b2716928856b0e930d36889c1db9ba3c5531eae5a750a626228f7fc53003 |
Hashes for fbgemm_gpu_cpu-0.5.0-cp38-cp38-manylinux2014_aarch64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 1d2a4a9d45e55c995398f2011397f837cfd84d251e63d0bb680fab326f79317a |
|
MD5 | 0b08efb0755ba6d077afdd8abe5e82b1 |
|
BLAKE2b-256 | 4eb77211c5fd9c99c81cae30c13e378ebfb8ad013956d316393ee5854e062499 |