# FBGEMM_GPU
[![FBGEMM_GPU CI](https://github.com/pytorch/FBGEMM/actions/workflows/fbgemm_gpu_ci.yml/badge.svg)](https://github.com/pytorch/FBGEMM/actions/workflows/fbgemm_gpu_ci.yml)
[![FBGEMM_GPU-CPU Nightly Build](https://github.com/pytorch/FBGEMM/actions/workflows/fbgemm_gpu_cpu_nightly.yml/badge.svg?event=schedule)](https://github.com/pytorch/FBGEMM/actions/workflows/fbgemm_gpu_cpu_nightly.yml)
[![FBGEMM_GPU-CUDA Nightly Build](https://github.com/pytorch/FBGEMM/actions/workflows/fbgemm_gpu_cuda_nightly.yml/badge.svg?event=schedule)](https://github.com/pytorch/FBGEMM/actions/workflows/fbgemm_gpu_cuda_nightly.yml)
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.0 and 11.8 in CI, and with PyTorch
packages (2.1+) that are built against those CUDA versions.
Only Intel/AMD CPUs with AVX2 extensions are currently supported.
See our [Documentation](docs/README.md) for more information.
## Installation
The full installation instructions
for the CUDA, ROCm, and CPU-only variants of FBGEMM_GPU can be found
[here](docs/InstallationInstructions.md). In addition, instructions for running
example tests and benchmarks can be found [here](docs/TestInstructions.md).
## 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](docs/BuildInstructions.md).
## Join the FBGEMM_GPU Community
For questions, support, news updates, or feature requests, please feel free to:
* File a ticket in [GitHub Issues](https://github.com/pytorch/FBGEMM/issues)
* Post a discussion in [GitHub Discussions](https://github.com/pytorch/FBGEMM/discussions)
* Reach out to us on the `#fbgemm` channel in [PyTorch Slack](https://bit.ly/ptslack)
For contributions, please see the [`CONTRIBUTING`](../CONTRIBUTING.md) file for
ways to help out.
## License
FBGEMM_GPU is BSD licensed, as found in the [`LICENSE`](../LICENSE) file.
Raw data
{
"_id": null,
"home_page": "https://github.com/pytorch/fbgemm",
"name": "fbgemm-gpu-cpu",
"maintainer": "",
"docs_url": null,
"requires_python": "",
"maintainer_email": "",
"keywords": "PyTorch,Recommendation Models,High Performance Computing,GPU,CUDA",
"author": "FBGEMM Team",
"author_email": "packages@pytorch.org",
"download_url": "",
"platform": null,
"description": "# FBGEMM_GPU\n\n[![FBGEMM_GPU CI](https://github.com/pytorch/FBGEMM/actions/workflows/fbgemm_gpu_ci.yml/badge.svg)](https://github.com/pytorch/FBGEMM/actions/workflows/fbgemm_gpu_ci.yml)\n[![FBGEMM_GPU-CPU Nightly Build](https://github.com/pytorch/FBGEMM/actions/workflows/fbgemm_gpu_cpu_nightly.yml/badge.svg?event=schedule)](https://github.com/pytorch/FBGEMM/actions/workflows/fbgemm_gpu_cpu_nightly.yml)\n[![FBGEMM_GPU-CUDA Nightly Build](https://github.com/pytorch/FBGEMM/actions/workflows/fbgemm_gpu_cuda_nightly.yml/badge.svg?event=schedule)](https://github.com/pytorch/FBGEMM/actions/workflows/fbgemm_gpu_cuda_nightly.yml)\n\nFBGEMM_GPU (FBGEMM GPU Kernels Library) is a collection of high-performance PyTorch\nGPU operator libraries for training and inference. The library provides efficient\ntable batched embedding bag, data layout transformation, and quantization supports.\n\nFBGEMM_GPU is currently tested with cuda 12.1.0 and 11.8 in CI, and with PyTorch\npackages (2.1+) that are built against those CUDA versions.\n\nOnly Intel/AMD CPUs with AVX2 extensions are currently supported.\n\nSee our [Documentation](docs/README.md) for more information.\n\n\n## Installation\n\nThe full installation instructions\nfor the CUDA, ROCm, and CPU-only variants of FBGEMM_GPU can be found\n[here](docs/InstallationInstructions.md). In addition, instructions for running\nexample tests and benchmarks can be found [here](docs/TestInstructions.md).\n\n\n## Build Instructions\n\nThis section is intended for FBGEMM_GPU developers only. The full build\ninstructions for the CUDA, ROCm, and CPU-only variants of FBGEMM_GPU can be\nfound [here](docs/BuildInstructions.md).\n\n\n## Join the FBGEMM_GPU Community\n\nFor questions, support, news updates, or feature requests, please feel free to:\n\n* File a ticket in [GitHub Issues](https://github.com/pytorch/FBGEMM/issues)\n* Post a discussion in [GitHub Discussions](https://github.com/pytorch/FBGEMM/discussions)\n* Reach out to us on the `#fbgemm` channel in [PyTorch Slack](https://bit.ly/ptslack)\n\nFor contributions, please see the [`CONTRIBUTING`](../CONTRIBUTING.md) file for\nways to help out.\n\n\n## License\n\nFBGEMM_GPU is BSD licensed, as found in the [`LICENSE`](../LICENSE) file.\n",
"bugtrack_url": null,
"license": "BSD-3",
"summary": "",
"version": "0.6.0",
"project_urls": {
"Homepage": "https://github.com/pytorch/fbgemm"
},
"split_keywords": [
"pytorch",
"recommendation models",
"high performance computing",
"gpu",
"cuda"
],
"urls": [
{
"comment_text": "",
"digests": {
"blake2b_256": "4810300aa05a92d8e9e7a745cf47838ea418b7f95a1a2371605a5b436b3bd45c",
"md5": "8242f59da42b63b05443f63517e54c5f",
"sha256": "b78601f57b24d7e62c24030b1662f0df5569b933eb471cbe966822a59d03dbbe"
},
"downloads": -1,
"filename": "fbgemm_gpu_cpu-0.6.0-cp310-cp310-manylinux2014_aarch64.whl",
"has_sig": false,
"md5_digest": "8242f59da42b63b05443f63517e54c5f",
"packagetype": "bdist_wheel",
"python_version": "cp310",
"requires_python": null,
"size": 2514326,
"upload_time": "2024-01-29T23:18:22",
"upload_time_iso_8601": "2024-01-29T23:18:22.415940Z",
"url": "https://files.pythonhosted.org/packages/48/10/300aa05a92d8e9e7a745cf47838ea418b7f95a1a2371605a5b436b3bd45c/fbgemm_gpu_cpu-0.6.0-cp310-cp310-manylinux2014_aarch64.whl",
"yanked": false,
"yanked_reason": null
},
{
"comment_text": "",
"digests": {
"blake2b_256": "a63867e6421d339dd169c36fbb4323f897fa5d286a5be97783d4f3cf446777ef",
"md5": "45e3830441c0d1d4fe977767e946fc18",
"sha256": "c824d8a3f18d8dba25b9b39cddd9cb9046e43bfa136999180c1b7cbb11695558"
},
"downloads": -1,
"filename": "fbgemm_gpu_cpu-0.6.0-cp310-cp310-manylinux2014_x86_64.whl",
"has_sig": false,
"md5_digest": "45e3830441c0d1d4fe977767e946fc18",
"packagetype": "bdist_wheel",
"python_version": "cp310",
"requires_python": null,
"size": 3381218,
"upload_time": "2024-01-29T23:15:34",
"upload_time_iso_8601": "2024-01-29T23:15:34.797988Z",
"url": "https://files.pythonhosted.org/packages/a6/38/67e6421d339dd169c36fbb4323f897fa5d286a5be97783d4f3cf446777ef/fbgemm_gpu_cpu-0.6.0-cp310-cp310-manylinux2014_x86_64.whl",
"yanked": false,
"yanked_reason": null
},
{
"comment_text": "",
"digests": {
"blake2b_256": "8c8fb94f6b7abb24051376e39ed13d1a7b4add418217957194a48612747667f0",
"md5": "f1206f39e78f6c71f3c19db4c37427ab",
"sha256": "42fe0788e07715b35c93351378d05471d482b5d81001b2be543dedaa458affb4"
},
"downloads": -1,
"filename": "fbgemm_gpu_cpu-0.6.0-cp311-cp311-manylinux2014_aarch64.whl",
"has_sig": false,
"md5_digest": "f1206f39e78f6c71f3c19db4c37427ab",
"packagetype": "bdist_wheel",
"python_version": "cp311",
"requires_python": null,
"size": 2514328,
"upload_time": "2024-01-29T23:18:41",
"upload_time_iso_8601": "2024-01-29T23:18:41.030869Z",
"url": "https://files.pythonhosted.org/packages/8c/8f/b94f6b7abb24051376e39ed13d1a7b4add418217957194a48612747667f0/fbgemm_gpu_cpu-0.6.0-cp311-cp311-manylinux2014_aarch64.whl",
"yanked": false,
"yanked_reason": null
},
{
"comment_text": "",
"digests": {
"blake2b_256": "55bfaa338d2d4d8d8e3fed0204f69620dd08d76e772b0ff620ab95b04461de87",
"md5": "8c906df535f4d206208ae88e29547511",
"sha256": "d921ffde365ea1857f900c0e6daf4630f5cc44ed3e84239395cdee4fa5948f6b"
},
"downloads": -1,
"filename": "fbgemm_gpu_cpu-0.6.0-cp311-cp311-manylinux2014_x86_64.whl",
"has_sig": false,
"md5_digest": "8c906df535f4d206208ae88e29547511",
"packagetype": "bdist_wheel",
"python_version": "cp311",
"requires_python": null,
"size": 3381221,
"upload_time": "2024-01-29T23:15:34",
"upload_time_iso_8601": "2024-01-29T23:15:34.359479Z",
"url": "https://files.pythonhosted.org/packages/55/bf/aa338d2d4d8d8e3fed0204f69620dd08d76e772b0ff620ab95b04461de87/fbgemm_gpu_cpu-0.6.0-cp311-cp311-manylinux2014_x86_64.whl",
"yanked": false,
"yanked_reason": null
},
{
"comment_text": "",
"digests": {
"blake2b_256": "818a1c7b4577533baedb502f944be2c9e65b202c94ef03669204f84ac4334150",
"md5": "6a9493d02b2aae5b41f5fa4c9725fa34",
"sha256": "203a98027df58f679076420fd964f25a400b4f7d3d4c0297fee0296d6640bc61"
},
"downloads": -1,
"filename": "fbgemm_gpu_cpu-0.6.0-cp312-cp312-manylinux2014_aarch64.whl",
"has_sig": false,
"md5_digest": "6a9493d02b2aae5b41f5fa4c9725fa34",
"packagetype": "bdist_wheel",
"python_version": "cp312",
"requires_python": null,
"size": 2514331,
"upload_time": "2024-01-29T23:18:30",
"upload_time_iso_8601": "2024-01-29T23:18:30.847687Z",
"url": "https://files.pythonhosted.org/packages/81/8a/1c7b4577533baedb502f944be2c9e65b202c94ef03669204f84ac4334150/fbgemm_gpu_cpu-0.6.0-cp312-cp312-manylinux2014_aarch64.whl",
"yanked": false,
"yanked_reason": null
},
{
"comment_text": "",
"digests": {
"blake2b_256": "58e8a02541fb64bd20f173d0f476141212ba0182e48c7f7a750fdb53f372a0e5",
"md5": "b3c1fc8a88cb6b8ec1f84fa963f6487d",
"sha256": "3beec17be632a1374bacc58f088ae72d49592817010c2314e42b49cd13fc86ae"
},
"downloads": -1,
"filename": "fbgemm_gpu_cpu-0.6.0-cp312-cp312-manylinux2014_x86_64.whl",
"has_sig": false,
"md5_digest": "b3c1fc8a88cb6b8ec1f84fa963f6487d",
"packagetype": "bdist_wheel",
"python_version": "cp312",
"requires_python": null,
"size": 3381220,
"upload_time": "2024-01-29T23:15:59",
"upload_time_iso_8601": "2024-01-29T23:15:59.106669Z",
"url": "https://files.pythonhosted.org/packages/58/e8/a02541fb64bd20f173d0f476141212ba0182e48c7f7a750fdb53f372a0e5/fbgemm_gpu_cpu-0.6.0-cp312-cp312-manylinux2014_x86_64.whl",
"yanked": false,
"yanked_reason": null
},
{
"comment_text": "",
"digests": {
"blake2b_256": "c7ee88b253dad59b5d3480dc06de936bce460aba4d33818e69ad7c45f025f570",
"md5": "82a367b85f1f3b4332ede440d10cf200",
"sha256": "857879281d90d2868c0ea771047cb88862db3f0b446646001a1ea65ead2dde73"
},
"downloads": -1,
"filename": "fbgemm_gpu_cpu-0.6.0-cp38-cp38-manylinux2014_aarch64.whl",
"has_sig": false,
"md5_digest": "82a367b85f1f3b4332ede440d10cf200",
"packagetype": "bdist_wheel",
"python_version": "cp38",
"requires_python": null,
"size": 2514318,
"upload_time": "2024-01-29T23:18:35",
"upload_time_iso_8601": "2024-01-29T23:18:35.083246Z",
"url": "https://files.pythonhosted.org/packages/c7/ee/88b253dad59b5d3480dc06de936bce460aba4d33818e69ad7c45f025f570/fbgemm_gpu_cpu-0.6.0-cp38-cp38-manylinux2014_aarch64.whl",
"yanked": false,
"yanked_reason": null
},
{
"comment_text": "",
"digests": {
"blake2b_256": "4ff7157b11a1f6fda85b544fb4752976360d0b778d99a055d7933f592fb7038b",
"md5": "759328d7a454d6a29e591bb212cf689d",
"sha256": "a1b14a4d8022d36279a7fc0d0ab7ad559567dad38f5035dcc3bb9c2aca02f8ad"
},
"downloads": -1,
"filename": "fbgemm_gpu_cpu-0.6.0-cp38-cp38-manylinux2014_x86_64.whl",
"has_sig": false,
"md5_digest": "759328d7a454d6a29e591bb212cf689d",
"packagetype": "bdist_wheel",
"python_version": "cp38",
"requires_python": null,
"size": 3381181,
"upload_time": "2024-01-29T23:15:39",
"upload_time_iso_8601": "2024-01-29T23:15:39.298949Z",
"url": "https://files.pythonhosted.org/packages/4f/f7/157b11a1f6fda85b544fb4752976360d0b778d99a055d7933f592fb7038b/fbgemm_gpu_cpu-0.6.0-cp38-cp38-manylinux2014_x86_64.whl",
"yanked": false,
"yanked_reason": null
},
{
"comment_text": "",
"digests": {
"blake2b_256": "7821f062d613c67830f9b881fc9793833e0bbee6c3b2eda7bc63be4b71ddde8e",
"md5": "0cd40143729498226f84c86c4e99048b",
"sha256": "900a6fba6feeaa8dda1f91f58e745d324f36b9d72b68bf346472cbdc98d5e2f2"
},
"downloads": -1,
"filename": "fbgemm_gpu_cpu-0.6.0-cp39-cp39-manylinux2014_aarch64.whl",
"has_sig": false,
"md5_digest": "0cd40143729498226f84c86c4e99048b",
"packagetype": "bdist_wheel",
"python_version": "cp39",
"requires_python": null,
"size": 2514319,
"upload_time": "2024-01-29T23:18:15",
"upload_time_iso_8601": "2024-01-29T23:18:15.547388Z",
"url": "https://files.pythonhosted.org/packages/78/21/f062d613c67830f9b881fc9793833e0bbee6c3b2eda7bc63be4b71ddde8e/fbgemm_gpu_cpu-0.6.0-cp39-cp39-manylinux2014_aarch64.whl",
"yanked": false,
"yanked_reason": null
},
{
"comment_text": "",
"digests": {
"blake2b_256": "814421b710b99702c5bd8a5ccad2fa9aa2381eaaa22c4d4f688f1965e1789b8d",
"md5": "6858691336dd98142402014e05fce7b6",
"sha256": "3319422ed7d33ea592631ad8e7e9a78989bdcf267ecb1f2c58f9939092bd0b23"
},
"downloads": -1,
"filename": "fbgemm_gpu_cpu-0.6.0-cp39-cp39-manylinux2014_x86_64.whl",
"has_sig": false,
"md5_digest": "6858691336dd98142402014e05fce7b6",
"packagetype": "bdist_wheel",
"python_version": "cp39",
"requires_python": null,
"size": 3381182,
"upload_time": "2024-01-29T23:15:59",
"upload_time_iso_8601": "2024-01-29T23:15:59.493168Z",
"url": "https://files.pythonhosted.org/packages/81/44/21b710b99702c5bd8a5ccad2fa9aa2381eaaa22c4d4f688f1965e1789b8d/fbgemm_gpu_cpu-0.6.0-cp39-cp39-manylinux2014_x86_64.whl",
"yanked": false,
"yanked_reason": null
}
],
"upload_time": "2024-01-29 23:18:22",
"github": true,
"gitlab": false,
"bitbucket": false,
"codeberg": false,
"github_user": "pytorch",
"github_project": "fbgemm",
"travis_ci": false,
"coveralls": false,
"github_actions": true,
"lcname": "fbgemm-gpu-cpu"
}