Skip to content

Commit 013d10c

Browse files
gchalumpmeta-codesync[bot]
authored andcommitted
Add CUDA 13.2 support to CI and release workflows (#5610)
Summary: Pull Request resolved: #5610 X-link: https://github.com/facebookresearch/FBGEMM/pull/2563 Add CUDA 13.2.0 to the FBGEMM GPU CI and release build matrices. Default publish version remains 13.0.2. Changes: - generate_ci_matrix.py: Add 13.2.0 to cuda_versions() for default and genai targets - fbgemm_gpu_release_cuda.yml: Add 13.2.0 to release options - fbgemm_gpu_release_genai.yml: Add 13.2.0 to genai release options and build/test matrices Differential Revision: D100229467
1 parent 6b72f97 commit 013d10c

3 files changed

Lines changed: 6 additions & 6 deletions

File tree

.github/scripts/generate_ci_matrix.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -304,10 +304,10 @@ def cuda_versions(self) -> List[str]:
304304
# FBGEMM HSTU is expensive, so conserve CI resources
305305
return ["12.8.1"]
306306
elif self.target == TARGET_GENAI:
307-
return ["12.6.3", "12.8.1", "12.9.1", "13.0.2"]
307+
return ["12.6.3", "12.8.1", "12.9.1", "13.0.2", "13.2.0"]
308308
else:
309309
# GenAI is unable to support 11.8.0 anymore as of https://github.com/pytorch/FBGEMM/pull/4138
310-
return ["12.6.3", "12.8.1", "12.9.1", "13.0.2"]
310+
return ["12.6.3", "12.8.1", "12.9.1", "13.0.2", "13.2.0"]
311311

312312
def rocm_versions(self) -> List[str]:
313313
if GitRepo.ref() == REFS_MAIN and GitRepo.event_name() == EVENT_NAME_PUSH:

.github/workflows/fbgemm_gpu_release_cuda.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@ on:
3434
description: CUDA Version to Use for Building Artifact
3535
type: choice
3636
required: false
37-
options: [ "12.6.3", "12.8.1", "12.9.1", "13.0.2" ]
37+
options: [ "12.6.3", "12.8.1", "12.9.1", "13.0.2", "13.2.0" ]
3838
default: "13.0.2"
3939
publish-to-pypi:
4040
description: Publish Artifact to PyPI

.github/workflows/fbgemm_gpu_release_genai.yml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@ on:
3434
description: CUDA Version to Use for Building Artifact
3535
type: choice
3636
required: false
37-
options: [ "12.6.3", "12.8.1", "12.9.1", "13.0.2" ]
37+
options: [ "12.6.3", "12.8.1", "12.9.1", "13.0.2", "13.2.0" ]
3838
default: "13.0.2"
3939
publish-to-pypi:
4040
description: Publish Artifact to PyPI
@@ -72,7 +72,7 @@ jobs:
7272
{ arch: x86, instance: "linux.12xlarge.memory" },
7373
]
7474
python-version: [ "3.10", "3.11", "3.12", "3.13", "3.14" ]
75-
cuda-version: [ "12.6.3", "12.8.1", "13.0.2" ]
75+
cuda-version: [ "12.6.3", "12.8.1", "13.0.2", "13.2.0" ]
7676

7777
steps:
7878
- name: Setup Build Container
@@ -146,7 +146,7 @@ jobs:
146146
{ arch: x86, instance: "linux.g5.4xlarge.nvidia.gpu" },
147147
]
148148
python-version: [ "3.10", "3.11", "3.12", "3.13", "3.14" ]
149-
cuda-version: [ "12.6.3", "12.8.1", "13.0.2" ]
149+
cuda-version: [ "12.6.3", "12.8.1", "13.0.2", "13.2.0" ]
150150
needs: build_artifact
151151

152152
steps:

0 commit comments

Comments
 (0)