From a0cf894a8c02c9b544910685e4429323cbe2dd52 Mon Sep 17 00:00:00 2001
From: Kannika Kabilar <kkabilar@tenstorrent.com>
Date: Wed, 15 Jan 2025 10:37:37 -0600
Subject: [PATCH] Added build-wheels to
 fast-dispatch-build-and-unit-tests-wrapper.yaml (#16638)

### Ticket
Link to Github Issue

https://github.com/tenstorrent/tt-metal/actions/runs/11943954436/job/33294528632

### Problem description
Provide context for the problem.
Added build-wheels to help with downloading the artifact in the workflow

### What's changed
Describe the approach used to solve the problem.
Summarize the changes made and its impact.

### Checklist
- [ ] Post commit CI passes
- [ ] Blackhole Post commit (if applicable)
- [ ] Model regression CI testing passes (if applicable)
- [ ] Device performance regression CI testing passes (if applicable)
- [ ] **(For models and ops writers)** Full [new
models](https://github.com/tenstorrent/tt-metal/actions/workflows/full-new-models-suite.yaml)
tests passes
- [ ] New/Existing tests provide coverage for changes
---
 ...dispatch-build-and-unit-tests-wrapper.yaml | 22 +++++++++++++++----
 1 file changed, 18 insertions(+), 4 deletions(-)

diff --git a/.github/workflows/fast-dispatch-build-and-unit-tests-wrapper.yaml b/.github/workflows/fast-dispatch-build-and-unit-tests-wrapper.yaml
index cfbaf686cd5..72df5019661 100644
--- a/.github/workflows/fast-dispatch-build-and-unit-tests-wrapper.yaml
+++ b/.github/workflows/fast-dispatch-build-and-unit-tests-wrapper.yaml
@@ -11,9 +11,23 @@ jobs:
     needs: build-docker-artifact
     uses: ./.github/workflows/build-artifact.yaml
     secrets: inherit
+  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]
+    uses: ./.github/workflows/_build-wheels-impl.yaml
+    with:
+      os: ${{ matrix.os }}
+      arch: ${{ matrix.arch }}
+      from-precompiled: true
+    secrets: inherit
   # FD Unit Tests
   fast-dispatch-unit-tests:
-    needs: build-artifact
+    needs: build-wheels
     secrets: inherit
     strategy:
       fail-fast: false
@@ -29,7 +43,7 @@ jobs:
       runner-label: ${{ matrix.test-group.runner-label}}
   # TTNN FD Unit tests
   ttnn-unit-tests:
-    needs: build-artifact
+    needs: build-wheels
     secrets: inherit
     strategy:
       fail-fast: false
@@ -46,7 +60,7 @@ jobs:
 
   # FD Model Tests
   models-unit-tests:
-    needs: build-artifact
+    needs: build-wheels
     secrets: inherit
     strategy:
       fail-fast: false
@@ -63,7 +77,7 @@ jobs:
 
   # FD C++ Unit Tests
   cpp-unit-tests:
-    needs: build-artifact
+    needs: build-wheels
     secrets: inherit
     strategy:
       fail-fast: false