Skip to content

Commit 60b8b5b

Browse files
committed
docker: Add separate images for CUDA 12
1 parent 77c6edc commit 60b8b5b

1 file changed

Lines changed: 14 additions & 7 deletions

File tree

.github/workflows/docker-bases.yml

Lines changed: 14 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -190,6 +190,7 @@ jobs:
190190
strategy:
191191
matrix:
192192
arch: [amd64, arm64]
193+
cuda: ['12', '13']
193194
include:
194195
- arch: amd64
195196
platform: linux/amd64
@@ -199,6 +200,12 @@ jobs:
199200
- arch: arm64
200201
platform: linux/arm64
201202
runner: ubuntu-24.04-arm
203+
- cuda: '12'
204+
nvhpc_version: 'nvhpc-25-7'
205+
extra_tag: '12'
206+
- cuda: '13'
207+
nvhpc_version: 'nvhpc'
208+
extra_tag: ''
202209

203210
steps:
204211
- name: Checkout devito
@@ -229,10 +236,10 @@ jobs:
229236
platforms: ${{ matrix.platform }}
230237
build-args: |
231238
arch=nvc
232-
ver=nvhpc-25-7
239+
ver=${{ matrix.nvhpc_version }}
233240
# Label (not tag) with runner name for traceability without changing image tags
234241
labels: builder-runner=${{ runner.name }}
235-
tags: "devitocodes/bases:nvidia-nvc-${{ matrix.arch }}"
242+
tags: "devitocodes/bases:nvidia-nvc${{ matrix.extra_tag }}-${{ matrix.arch }}"
236243

237244
- name: NVCC image
238245
uses: docker/build-push-action@v6
@@ -244,9 +251,9 @@ jobs:
244251
platforms: ${{ matrix.platform }}
245252
build-args: |
246253
arch=nvcc
247-
ver=nvhpc-25-7
254+
ver=${{ matrix.nvhpc_version }}
248255
labels: builder-runner=${{ runner.name }}
249-
tags: "devitocodes/bases:nvidia-nvcc-${{ matrix.arch }}"
256+
tags: "devitocodes/bases:nvidia-nvcc${{ matrix.extra_tag }}-${{ matrix.arch }}"
250257

251258
- name: NVC host image
252259
uses: docker/build-push-action@v6
@@ -258,9 +265,9 @@ jobs:
258265
platforms: ${{ matrix.platform }}
259266
build-args: |
260267
arch=nvc-host
261-
ver=nvhpc-25-7
268+
ver=${{ matrix.nvhpc_version }}
262269
labels: builder-runner=${{ runner.name }}
263-
tags: "devitocodes/bases:cpu-nvc-${{ matrix.arch }}"
270+
tags: "devitocodes/bases:cpu-nvc${{ matrix.extra_tag }}-${{ matrix.arch }}"
264271

265272
deploy-nvidia-bases-manifest:
266273
if: inputs.nvidia
@@ -272,7 +279,7 @@ jobs:
272279

273280
strategy:
274281
matrix:
275-
final_tag: [nvidia-nvc, nvidia-nvcc, cpu-nvc]
282+
final_tag: [nvidia-nvc, nvidia-nvcc, cpu-nvc, nvidia-nvc12, nvidia-nvcc12, cpu-nvc12]
276283

277284
steps:
278285
- name: Set up Docker Buildx

0 commit comments

Comments
 (0)