From 1dbdf75654e737c45ff33aa4ebf3090e088ba5c9 Mon Sep 17 00:00:00 2001
From: Usman Aziz <uaziz@tenstorrent.com>
Date: Mon, 13 Jan 2025 09:37:46 -0500
Subject: [PATCH] Copy pre-built env from build to all test runners (#182)

---
 .github/workflows/run-build.yml       | 2 +-
 .github/workflows/run-model-tests.yml | 2 +-
 .github/workflows/run-tests.yml       | 9 ++++++---
 3 files changed, 8 insertions(+), 5 deletions(-)

diff --git a/.github/workflows/run-build.yml b/.github/workflows/run-build.yml
index 7ba2bf36..ca7f801a 100644
--- a/.github/workflows/run-build.yml
+++ b/.github/workflows/run-build.yml
@@ -116,7 +116,7 @@ jobs:
       working-directory: ${{ steps.strings.outputs.install-output-dir }}
       run: |
         mkdir env
-        cp -r ${{ steps.strings.outputs.work-dir }}/env env
+        cp -r ${{ steps.strings.outputs.work-dir }}/env/* env
 
     - name: 'Tar install directory'
       shell: bash
diff --git a/.github/workflows/run-model-tests.yml b/.github/workflows/run-model-tests.yml
index ecb17405..328e31a5 100644
--- a/.github/workflows/run-model-tests.yml
+++ b/.github/workflows/run-model-tests.yml
@@ -80,7 +80,7 @@ jobs:
       working-directory: ${{ steps.strings.outputs.install-output-dir }}
       run: |
         cp -r ${{ steps.strings.outputs.install-output-dir }}/tt-metal/* ${{ steps.strings.outputs.work-dir }}/third_party/tt-mlir/src/tt-mlir/third_party/tt-metal/src/tt-metal
-        cp -r ${{ steps.strings.outputs.install-output-dir }}/env/env/* ${{ steps.strings.outputs.work-dir }}/env
+        cp -r ${{ steps.strings.outputs.install-output-dir }}/env/* ${{ steps.strings.outputs.work-dir }}/env
 
     - name: Run Model Tests
       shell: bash
diff --git a/.github/workflows/run-tests.yml b/.github/workflows/run-tests.yml
index cb29491f..1421bff4 100644
--- a/.github/workflows/run-tests.yml
+++ b/.github/workflows/run-tests.yml
@@ -73,16 +73,19 @@ jobs:
       working-directory: ${{ steps.strings.outputs.install-output-dir }}
       run: tar xvf artifact.tar
 
-    - name: make tt-metal directory
+    - name: make tt-metal and env directories
       shell: bash
       working-directory: ${{ steps.strings.outputs.install-output-dir }}
-      run: mkdir -p ${{ steps.strings.outputs.work-dir }}/third_party/tt-mlir/src/tt-mlir/third_party/tt-metal/src/tt-metal
+      run: |
+        mkdir -p ${{ steps.strings.outputs.work-dir }}/third_party/tt-mlir/src/tt-mlir/third_party/tt-metal/src/tt-metal
+        mkdir -p ${{ steps.strings.outputs.work-dir }}/env
 
-    - name: copy tt-metal dirs
+    - name: copy tt-metal and env dirs
       shell: bash
       working-directory: ${{ steps.strings.outputs.install-output-dir }}
       run: |
         cp -r ${{ steps.strings.outputs.install-output-dir }}/tt-metal/* ${{ steps.strings.outputs.work-dir }}/third_party/tt-mlir/src/tt-mlir/third_party/tt-metal/src/tt-metal
+        cp -r ${{ steps.strings.outputs.install-output-dir }}/env/* ${{ steps.strings.outputs.work-dir }}/env
 
     - name: Run PyTorch Unit tests
       shell: bash