From 3d3b51efea59c46c0ac7e863b16a54f908a631d4 Mon Sep 17 00:00:00 2001
From: Dimitri Gnidash <dimitri@tenstorrent.com>
Date: Mon, 5 Aug 2024 14:23:06 -0400
Subject: [PATCH] Make sure the images are split by arch

---
 .github/workflows/build-artifact.yaml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/.github/workflows/build-artifact.yaml b/.github/workflows/build-artifact.yaml
index aba3136230b6..b7c4c6c8020b 100644
--- a/.github/workflows/build-artifact.yaml
+++ b/.github/workflows/build-artifact.yaml
@@ -20,7 +20,7 @@ jobs:
       TT_METAL_ENV: ${{ vars.TT_METAL_ENV }}
       ARCH_NAME: ${{ matrix.arch }}
       IMAGE_TAG: dev-${GITHUB_ACTOR}.${{ matrix.arch }}
-      TT_METAL_DOCKER_IMAGE: tt-metalium/ubuntu-20.04-amd64
+      TT_METAL_DOCKER_IMAGE: tt-metalium/ubuntu-20.04-amd64/${{ matrix.arch }}
       TT_METAL_DOCKER_IMAGE_TAG: ${{ github.sha }}
     environment: dev
     runs-on: build-docker