Skip to content

Commit

Permalink
Add CUDA 12.2 (#634)
Browse files Browse the repository at this point in the history
Add CUDA 12.2 images to the build matrix

References:

* rapidsai/build-planning#6
* rapidsai/build-planning#7
* rapidsai/build-planning#8

Authors:
  - https://github.com/jakirkham
  - Ray Douglass (https://github.com/raydouglass)

Approvers:
  - Bradley Dice (https://github.com/bdice)
  - Ray Douglass (https://github.com/raydouglass)

URL: #634
  • Loading branch information
jakirkham authored Feb 26, 2024
1 parent 3412754 commit 32a60e1
Show file tree
Hide file tree
Showing 3 changed files with 5 additions and 2 deletions.
2 changes: 1 addition & 1 deletion .github/workflows/build-image.yml
Original file line number Diff line number Diff line change
Expand Up @@ -142,7 +142,7 @@ jobs:
RAPIDS_VER=${{ inputs.RAPIDS_VER }}
tags: ${{ inputs.RAFT_ANN_BENCH_DATASETS_TAG }}-${{ matrix.ARCH }}
- name: Build RAFT ANN Benchmarks CPU image
if: inputs.CUDA_VER == '12.0.1' # we don't need to build CPU packages for different CUDA versions.
if: inputs.CUDA_VER == '12.2.2' # we don't need to build CPU packages for different CUDA versions.
uses: docker/build-push-action@v4
with:
context: context
Expand Down
4 changes: 3 additions & 1 deletion matrix-test.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -3,9 +3,11 @@
pull-request:
- { CUDA_VER: '11.8', ARCH: 'amd64', PYTHON_VER: '3.9', GPU: 'v100', DRIVER: 'earliest' }
- { CUDA_VER: '12.0', ARCH: 'amd64', PYTHON_VER: '3.10', GPU: 'v100', DRIVER: 'latest' }
- { CUDA_VER: '12.0', ARCH: 'arm64', PYTHON_VER: '3.10', GPU: 'a100', DRIVER: 'latest' }
- { CUDA_VER: '12.2', ARCH: 'arm64', PYTHON_VER: '3.10', GPU: 'a100', DRIVER: 'latest' }
branch:
- { CUDA_VER: '11.8', ARCH: 'amd64', PYTHON_VER: '3.9', GPU: 'v100', DRIVER: 'earliest' }
- { CUDA_VER: '11.8', ARCH: 'amd64', PYTHON_VER: '3.9', GPU: 'v100', DRIVER: 'latest' }
- { CUDA_VER: '12.0', ARCH: 'amd64', PYTHON_VER: '3.10', GPU: 'v100', DRIVER: 'latest' }
- { CUDA_VER: '12.0', ARCH: 'arm64', PYTHON_VER: '3.10', GPU: 'a100', DRIVER: 'latest' }
- { CUDA_VER: '12.2', ARCH: 'amd64', PYTHON_VER: '3.10', GPU: 'v100', DRIVER: 'latest' }
- { CUDA_VER: '12.2', ARCH: 'arm64', PYTHON_VER: '3.10', GPU: 'a100', DRIVER: 'latest' }
1 change: 1 addition & 0 deletions matrix.yaml
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
CUDA_VER: # Should be `<major>.<minor>.<patch>` (e.g. `11.2.2`)
- "11.8.0"
- "12.0.1"
- "12.2.2"
PYTHON_VER:
- "3.9"
- "3.10"
Expand Down

0 comments on commit 32a60e1

Please sign in to comment.