Update CI Tags (#65)

mila-iqia · Mar 8, 2023 · 3acc0b3 · 3acc0b3
1 parent f2a472b
commit 3acc0b3
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 13 deletions.
diff --git a/.github/workflows/docker-run.yml b/.github/workflows/docker-run.yml
@@ -7,16 +7,17 @@ on:
 
 env:
   REGISTRY: ghcr.io
+  IMAGE_NAME: ${{ github.repository }}
+  IMAGE_PATH: ghcr.io/mila-iqia/milabench:cuda-nightly
 
 jobs:
-  build-image:
-    runs-on: [self-hosted, gpu]
+  docker-run:
+    runs-on: [self-hosted, cuda]
 
     strategy:
       matrix:
         include:
           - arch: cuda
-            config: standard-cuda.yaml
 
     permissions:
       contents: read
@@ -25,13 +26,6 @@ jobs:
       IMAGE_NAME: "ghcr.io/mila-iqia/milabench:${{ matrix.arch }}-nightly"
 
     steps:
-      - name: login
-        uses: docker/login-action@v2
-        with:
-          registry: ${{ env.REGISTRY }}
-          username: ${{ github.actor }}
-          password: ${{ secrets.GITHUB_TOKEN }}
-
       - name: pull
         run: |
           docker pull $IMAGE_NAME
@@ -54,4 +48,4 @@ jobs:
           python -m pip install -U poetry
           poetry lock --no-update
           poetry install
-          milabench summary $OUTPUT/
+          milabench summary $OUTPUT
diff --git a/.github/workflows/docker.yml b/.github/workflows/docker.yml
@@ -22,7 +22,7 @@ permissions:
 
 jobs:
   build-image:
-    runs-on: [self-hosted, gpu]
+    runs-on: [self-hosted, cuda]
 
     # define build arguments
     strategy:

diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
@@ -21,7 +21,7 @@ on:
 
 jobs:
   tests:
-    runs-on: [self-hosted, gpu]
+    runs-on: [self-hosted, cuda]
     defaults:
       run:
         shell: bash -el {0}