From 3acc0b3b1288b1053e4179b8c08c10a4b4ad4afe Mon Sep 17 00:00:00 2001
From: Setepenre <pierre.delaunay.tr@gmail.com>
Date: Wed, 8 Mar 2023 10:47:32 -0500
Subject: [PATCH] Update CI Tags (#65)

---
 .github/workflows/docker-run.yml | 16 +++++-----------
 .github/workflows/docker.yml     |  2 +-
 .github/workflows/tests.yml      |  2 +-
 3 files changed, 7 insertions(+), 13 deletions(-)

diff --git a/.github/workflows/docker-run.yml b/.github/workflows/docker-run.yml
index 60750615c..1f5788061 100644
--- 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
index 3764b3688..3e11c8f21 100644
--- 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
index 3006c6fa0..5479e8ae4 100644
--- 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}