Skip to content

Commit ac76e88

Browse files
authored
build for aarch64-pytorch-cuda11 as well (for GH100 and simplified CI config)
1 parent 3817d82 commit ac76e88

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

.github/workflows/docker.yml

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -260,6 +260,18 @@ jobs:
260260
needs: x86_64-scipy
261261
if: ${{ !contains(github.event.pull_request.title, '[FAST_BUILD]') }}
262262

263+
aarch64-pytorch-cuda11:
264+
uses: ./.github/workflows/docker-build-test-upload.yml
265+
with:
266+
parent-image: scipy-notebook
267+
image: pytorch-notebook
268+
variant: cuda11
269+
platform: aarch64
270+
runs-on: ubuntu-24.04-arm
271+
timeout-minutes: 25
272+
needs: aarch64-scipy
273+
if: ${{ !contains(github.event.pull_request.title, '[FAST_BUILD]') }}
274+
263275
x86_64-pytorch-cuda11:
264276
uses: ./.github/workflows/docker-build-test-upload.yml
265277
with:

0 commit comments

Comments
 (0)