From 03b0392334693ca0ea9e3d159d3905b668f62095 Mon Sep 17 00:00:00 2001
From: uazizTT <uaziz@tenstorrent.com>
Date: Wed, 11 Dec 2024 15:34:10 -0500
Subject: [PATCH] Copy over the env from build machine to reduce runtime of
 test runners.

---
 .github/workflows/generate-model-report.yml | 2 +-
 .github/workflows/nightly-tests.yml         | 2 +-
 .github/workflows/nightly-uplift.yml        | 2 +-
 .github/workflows/run-build.yml             | 7 +++++++
 .github/workflows/run-model-tests.yml       | 9 ++++++---
 5 files changed, 16 insertions(+), 6 deletions(-)

diff --git a/.github/workflows/generate-model-report.yml b/.github/workflows/generate-model-report.yml
index f3e7c5a1..fcfa9302 100644
--- a/.github/workflows/generate-model-report.yml
+++ b/.github/workflows/generate-model-report.yml
@@ -85,7 +85,7 @@ jobs:
         rm artifact.tar
         for a in `ls -1 *.tar`; do tar -xvf $a; rm $a; done
 
-    - name: Display structure of downloaded file
+    - name: Display structure of downloaded dir
       working-directory: ${{ steps.strings.outputs.test-output-dir }}
       run: |
         ls -R ${{ steps.strings.outputs.test-output-dir }}
diff --git a/.github/workflows/nightly-tests.yml b/.github/workflows/nightly-tests.yml
index 1aaab4d5..d173ef6b 100644
--- a/.github/workflows/nightly-tests.yml
+++ b/.github/workflows/nightly-tests.yml
@@ -1,6 +1,6 @@
 # This workflow automates running model tests.
 
-name: Nighty Tests
+name: Nightly Tests
 
 on:
   schedule:
diff --git a/.github/workflows/nightly-uplift.yml b/.github/workflows/nightly-uplift.yml
index e80e7952..9ee97d8c 100644
--- a/.github/workflows/nightly-uplift.yml
+++ b/.github/workflows/nightly-uplift.yml
@@ -1,7 +1,7 @@
 # This workflow automates creation of uplift pull requests.
 # Uplift PR is created daily to uplift the submodule to the latest version.
 
-name: Nighty Uplift
+name: Nightly Uplift
 
 on:
   schedule:
diff --git a/.github/workflows/run-build.yml b/.github/workflows/run-build.yml
index d536787e..1ad34c06 100644
--- a/.github/workflows/run-build.yml
+++ b/.github/workflows/run-build.yml
@@ -164,6 +164,13 @@ jobs:
         cp -r ${{ steps.strings.outputs.work-dir }}/third_party/tt-mlir/src/tt-mlir/third_party/tt-metal/src/tt-metal/ttnn tt-metal/
         cp -r ${{ steps.strings.outputs.work-dir }}/third_party/tt-mlir/src/tt-mlir/third_party/tt-metal/src/tt-metal/runtime tt-metal/
 
+    - name: Copy venv directory
+      shell: bash
+      working-directory: ${{ steps.strings.outputs.install-output-dir }}
+      run: |
+        mkdir env
+        cp -r ${{ steps.strings.outputs.work-dir }}/env env
+
     - name: 'Tar install directory'
       shell: bash
       working-directory: ${{ steps.strings.outputs.install-output-dir }}
diff --git a/.github/workflows/run-model-tests.yml b/.github/workflows/run-model-tests.yml
index 5cc95ab8..862589ac 100644
--- a/.github/workflows/run-model-tests.yml
+++ b/.github/workflows/run-model-tests.yml
@@ -69,16 +69,19 @@ jobs:
       working-directory: ${{ steps.strings.outputs.install-output-dir }}
       run: tar xvf artifact.tar
 
-    - name: make tt-metal directory
+    - name: make 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/env/* ${{ steps.strings.outputs.work-dir }}/env
 
     - name: Run Model Tests
       shell: bash