Copy over the env from build machine to reduce runtime of test runners.

tenstorrent · Dec 12, 2024 · 03b0392 · 03b0392
1 parent 6ff28b5
commit 03b0392
Show file tree

Hide file tree

Showing 5 changed files with 16 additions and 6 deletions.
diff --git 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
@@ -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
@@ -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
@@ -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
@@ -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