Pass build argument to select base image

tenstorrent · Feb 2, 2025 · f76612b · f76612b
1 parent b2f74a9
commit f76612b
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/.github/workflows/build-docker-artifact.yaml b/.github/workflows/build-docker-artifact.yaml
@@ -26,7 +26,6 @@ jobs:
       SILENT: 0
       VERBOSE: 1
       TT_METAL_DOCKER_IMAGE: ${{ inputs.os }}
-      IMAGE: tt-metalium/ubuntu-20.04-amd64
       DOCKERFILE: ubuntu-20.04-amd64
     runs-on:
       - build-docker
@@ -66,7 +65,9 @@ jobs:
           target: dev
           push: true
           tags: ${{ env.TT_METAL_DOCKER_IMAGE_TAG}}
-          no-cache: true
+          build-args: UBUNTU_VERSION=${{ fromJSON('["' + join(split(inputs.os, '-'), '","') + '"]')[1] }}
+          cache-from: type=registry,ref=${{ env.TT_METAL_REF_IMAGE_TAG }}
+          cache-to: type=inline
           pull: true
       - name: Tag Docker main image as current image
         if: steps.changed-files-specific.outputs.any_changed != 'true'