#8082: add github action to load docker image artifact

tenstorrent · May 10, 2024 · 5287d23 · 5287d23
1 parent 5719ab8
commit 5287d23
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 7 deletions.
diff --git a/.github/actions/load-docker-image/action.yml b/.github/actions/load-docker-image/action.yml
@@ -0,0 +1,23 @@
+name: "Load Docker Image"
+description: "Downloads and loads Docker image artifact"
+inputs:
+  artifact-name:
+    description: 'The name of the Docker image artifact to download.'
+    required: true
+    default: 'docker_image.tar.gz'
+  path:
+    description: 'The path where the Docker image artifact will be downloaded.'
+    required: true
+    default: '/tmp'
+runs:
+  using: "composite"
+  steps:
+    - name: Download Docker image
+      uses: actions/download-artifact@v2
+      with:
+        name: ${{ inputs.artifact-name }}
+        path: ${{ inputs.path }}
+
+    - name: Load Docker image
+      run: |
+        gunzip -c ${{ inputs.path }}/${{ inputs.artifact-name }} | docker load
diff --git a/.github/workflows/build-artifact.yaml b/.github/workflows/build-artifact.yaml
@@ -48,14 +48,11 @@ jobs:
         run: |
           echo "TT_METAL_HOME=$(pwd)" >> $GITHUB_ENV
       - uses: ./.github/actions/install-python-deps
-      - name: Download Docker image
-        uses: actions/download-artifact@v2
+      - name: Load Docker Image
+        uses: ./.github/actions/load-docker-image
         with:
-          name: docker_image.tar.gz 
-          path: /tmp
-      - name: Load Docker image
-        run: |
-          gunzip -c /tmp/docker_image.tar.gz | docker load
+          artifact-name: 'docker_image.tar.gz'
+          path: '/tmp'
       - name: Build tt-metal cmake build 
         run: |
           ./scripts/docker/run_docker_cmd.sh -d "cmake -B build"