Install wheel in venv

tenstorrent · Dec 9, 2024 · 874f9b0 · 874f9b0
1 parent 74c1f80
commit 874f9b0
Show file tree

Hide file tree

Showing 5 changed files with 22 additions and 5 deletions.
diff --git a/.github/workflows/all-post-commit-workflows.yaml b/.github/workflows/all-post-commit-workflows.yaml
@@ -90,7 +90,7 @@ jobs:
       runner-label: ${{ matrix.test-group.runner-label }}
   # Slow Dispatch Unit Tests
   sd-unit-tests:
-    needs: build-artifact
+    needs: build-wheels
     secrets: inherit
     strategy:
       fail-fast: false
@@ -155,7 +155,7 @@ jobs:
       runner-label: ${{ matrix.test-group.runner-label }}
   # FD C++ Unit Tests
   cpp-unit-tests:
-    needs: build-artifact
+    needs: build-wheels
     secrets: inherit
     strategy:
       fail-fast: false

diff --git a/.github/workflows/build-and-unit-tests-wrapper.yaml b/.github/workflows/build-and-unit-tests-wrapper.yaml
@@ -7,9 +7,16 @@ jobs:
   build-artifact:
     uses: ./.github/workflows/build-artifact.yaml
     secrets: inherit
-
-  sd-unit-tests:
+  build-wheels:
     needs: build-artifact
+    strategy:
+      matrix:
+        # Since pre-compiled builds only run on 20.04, we can only test on 20.04 for now
+        # The full 22.04 flow can be tested without precompiled
+        os: [ubuntu-20.04]
+        arch: [grayskull, wormhole_b0]
+  sd-unit-tests:
+    needs: build-wheels
     secrets: inherit
     strategy:
       fail-fast: false

diff --git a/.github/workflows/build-and-unit-tests.yaml b/.github/workflows/build-and-unit-tests.yaml
@@ -69,6 +69,7 @@ jobs:
             -e TT_METAL_HOME=${{ github.workspace }}
             -e TT_METAL_SLOW_DISPATCH_MODE=1
             -e LD_LIBRARY_PATH=${{ github.workspace }}/build/lib
+          install_wheel: true
           run_args: |
             ./tests/scripts/run_tests.sh --tt-arch ${{ inputs.arch }} --pipeline-type post_commit --dispatch-mode slow
       - uses: ./.github/actions/slack-report

diff --git a/.github/workflows/cpp-post-commit-wrapper.yaml b/.github/workflows/cpp-post-commit-wrapper.yaml
@@ -11,8 +11,16 @@ jobs:
   build-artifact:
     uses: ./.github/workflows/build-artifact.yaml
     secrets: inherit
-  cpp-unit-tests:
+  build-wheels:
     needs: build-artifact
+    strategy:
+      matrix:
+        # Since pre-compiled builds only run on 20.04, we can only test on 20.04 for now
+        # The full 22.04 flow can be tested without precompiled
+        os: [ubuntu-20.04]
+        arch: [grayskull, wormhole_b0]
+  cpp-unit-tests:
+    needs: build-wheels
     secrets: inherit
     strategy:
       fail-fast: false

diff --git a/.github/workflows/cpp-post-commit.yaml b/.github/workflows/cpp-post-commit.yaml
@@ -83,6 +83,7 @@ jobs:
             -e TT_METAL_HOME=${{ github.workspace }}
             -e ARCH_NAME=${{ inputs.arch }}
             -e LD_LIBRARY_PATH=${{ github.workspace }}/build/lib
+          install_wheel: true
           run_args: |
             ${{ matrix.test-group.cmd }}
       - uses: ./.github/actions/slack-report