# FBGEMM_GPU
[![FBGEMM_GPU-CPU CI](https://github.com/pytorch/FBGEMM/actions/workflows/fbgemm_gpu_ci_cpu.yml/badge.svg)](https://github.com/pytorch/FBGEMM/actions/workflows/fbgemm_gpu_ci_cpu.yml)
[![FBGEMM_GPU-CUDA CI](https://github.com/pytorch/FBGEMM/actions/workflows/fbgemm_gpu_ci_cuda.yml/badge.svg)](https://github.com/pytorch/FBGEMM/actions/workflows/fbgemm_gpu_ci_cuda.yml)
[![FBGEMM_GPU-ROCm CI](https://github.com/pytorch/FBGEMM/actions/workflows/fbgemm_gpu_ci_rocm.yml/badge.svg)](https://github.com/pytorch/FBGEMM/actions/workflows/fbgemm_gpu_ci_rocm.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 and 11.8 in CI, and with PyTorch
packages (2.1+) that are built against those CUDA versions.
See the full [Documentation](https://pytorch.org/FBGEMM) 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](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-genai",
"maintainer": null,
"docs_url": null,
"requires_python": null,
"maintainer_email": null,
"keywords": "PyTorch, Recommendation Models, High Performance Computing, GPU, CUDA",
"author": "FBGEMM Team",
"author_email": "packages@pytorch.org",
"download_url": null,
"platform": null,
"description": "# FBGEMM_GPU\n\n[![FBGEMM_GPU-CPU CI](https://github.com/pytorch/FBGEMM/actions/workflows/fbgemm_gpu_ci_cpu.yml/badge.svg)](https://github.com/pytorch/FBGEMM/actions/workflows/fbgemm_gpu_ci_cpu.yml)\n[![FBGEMM_GPU-CUDA CI](https://github.com/pytorch/FBGEMM/actions/workflows/fbgemm_gpu_ci_cuda.yml/badge.svg)](https://github.com/pytorch/FBGEMM/actions/workflows/fbgemm_gpu_ci_cuda.yml)\n[![FBGEMM_GPU-ROCm CI](https://github.com/pytorch/FBGEMM/actions/workflows/fbgemm_gpu_ci_rocm.yml/badge.svg)](https://github.com/pytorch/FBGEMM/actions/workflows/fbgemm_gpu_ci_rocm.yml)\n\nFBGEMM_GPU (FBGEMM GPU Kernels Library) is a collection of high-performance\nPyTorch GPU operator libraries for training and inference. The library provides\nefficient table batched embedding bag, data layout transformation, and\nquantization supports.\n\nFBGEMM_GPU is currently tested with CUDA 12.1 and 11.8 in CI, and with PyTorch\npackages (2.1+) that are built against those CUDA versions.\n\nSee the full [Documentation](https://pytorch.org/FBGEMM) for more information\non building, installing, and developing with FBGEMM_GPU, as well as the most\nup-to-date support matrix for this library.\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": null,
"version": "0.7.2",
"project_urls": {
"Homepage": "https://github.com/pytorch/fbgemm"
},
"split_keywords": [
"pytorch",
" recommendation models",
" high performance computing",
" gpu",
" cuda"
],
"urls": [
{
"comment_text": "",
"digests": {
"blake2b_256": "543eed66e8d2c66524cb171a92d3a2637789ead21f51a5826412adbf0c899fd8",
"md5": "a824c02472a505903b1488a35a65865a",
"sha256": "5e6dff4a79339f27241f247d77462e5fceb7982a32726bfc266ea15fed658327"
},
"downloads": -1,
"filename": "fbgemm_gpu_genai-0.7.2-cp310-cp310-manylinux1_x86_64.whl",
"has_sig": false,
"md5_digest": "a824c02472a505903b1488a35a65865a",
"packagetype": "bdist_wheel",
"python_version": "cp310",
"requires_python": null,
"size": 2179904,
"upload_time": "2024-07-21T05:39:11",
"upload_time_iso_8601": "2024-07-21T05:39:11.776552Z",
"url": "https://files.pythonhosted.org/packages/54/3e/ed66e8d2c66524cb171a92d3a2637789ead21f51a5826412adbf0c899fd8/fbgemm_gpu_genai-0.7.2-cp310-cp310-manylinux1_x86_64.whl",
"yanked": false,
"yanked_reason": null
},
{
"comment_text": "",
"digests": {
"blake2b_256": "4eff6260d10d4b00177579300fec8a2b26bd8391ef6fd149df23086496c06916",
"md5": "509ab15d32289b550a8e76cc46f75d05",
"sha256": "09dc84d9d95d6e9d63bc23470c6f1e2dfd83de86e631d19d04230d08a77030ae"
},
"downloads": -1,
"filename": "fbgemm_gpu_genai-0.7.2-cp311-cp311-manylinux1_x86_64.whl",
"has_sig": false,
"md5_digest": "509ab15d32289b550a8e76cc46f75d05",
"packagetype": "bdist_wheel",
"python_version": "cp311",
"requires_python": null,
"size": 2179916,
"upload_time": "2024-07-21T06:07:44",
"upload_time_iso_8601": "2024-07-21T06:07:44.501845Z",
"url": "https://files.pythonhosted.org/packages/4e/ff/6260d10d4b00177579300fec8a2b26bd8391ef6fd149df23086496c06916/fbgemm_gpu_genai-0.7.2-cp311-cp311-manylinux1_x86_64.whl",
"yanked": false,
"yanked_reason": null
},
{
"comment_text": "",
"digests": {
"blake2b_256": "f6a09567077b7c800e19b026bb1abcb0bf48c0dfd79c8ba6c3d1392ea9558ec4",
"md5": "4ec1c5c57f26249b0e7d569323749d90",
"sha256": "fef6c94755bff9de49f8d2a2d0c1e05e5c7056d03c4989a32b6cf272e539c767"
},
"downloads": -1,
"filename": "fbgemm_gpu_genai-0.7.2-cp312-cp312-manylinux1_x86_64.whl",
"has_sig": false,
"md5_digest": "4ec1c5c57f26249b0e7d569323749d90",
"packagetype": "bdist_wheel",
"python_version": "cp312",
"requires_python": null,
"size": 2179918,
"upload_time": "2024-07-21T06:38:49",
"upload_time_iso_8601": "2024-07-21T06:38:49.498037Z",
"url": "https://files.pythonhosted.org/packages/f6/a0/9567077b7c800e19b026bb1abcb0bf48c0dfd79c8ba6c3d1392ea9558ec4/fbgemm_gpu_genai-0.7.2-cp312-cp312-manylinux1_x86_64.whl",
"yanked": false,
"yanked_reason": null
}
],
"upload_time": "2024-07-21 05:39:11",
"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-genai"
}