From 0ef7e21da2e30e5e5a9bfab0c04c4e7283ca7059 Mon Sep 17 00:00:00 2001
From: Michael Chiou <mchiou@tenstorrent.com>
Date: Wed, 15 May 2024 15:04:38 -0700
Subject: [PATCH] #8535: update ttnn sweep workflow to use build artifacts

---
 .github/workflows/ttnn-run-sweeps.yaml | 13 ++++++++++---
 1 file changed, 10 insertions(+), 3 deletions(-)

diff --git a/.github/workflows/ttnn-run-sweeps.yaml b/.github/workflows/ttnn-run-sweeps.yaml
index 649a6a518c9..5fd90328fcb 100644
--- a/.github/workflows/ttnn-run-sweeps.yaml
+++ b/.github/workflows/ttnn-run-sweeps.yaml
@@ -7,7 +7,11 @@ on:
   workflow_call:
 
 jobs:
+  build-artifact:
+    uses: ./.github/workflows/build-artifact.yaml
+    secrets: inherit
   ttnn-sweeps:
+    needs: build-artifact
     strategy:
       # Do not fail-fast because we need to ensure all tests go to completion
       # so we try not to get hanging machines
@@ -24,9 +28,12 @@ jobs:
       - name: Set up dynamic env vars for build
         run: |
           echo "TT_METAL_HOME=$(pwd)" >> $GITHUB_ENV
-      - name: Build tt-metal and libs
-        run: |
-          PYTHON_ENV_DIR=$(pwd)/build/python_env ./build_metal.sh
+      - uses: actions/download-artifact@v4
+        with:
+          name: TTMetal_build_${{ matrix.runner-info.arch }}
+      - name: Extract files
+        run: tar -xvf ttm_${{ matrix.runner-info.arch }}.tar
+      - uses: ./.github/actions/install-python-deps
       - name: Run ttnn sweeps
         timeout-minutes: 30
         run: ./tests/scripts/run_tests.sh --tt-arch $ARCH_NAME --pipeline-type ttnn_sweeps