From dc73f7fcfb71c96917346e11ba396f461dbcdc5b Mon Sep 17 00:00:00 2001
From: Vladimir Milosevic <vmilosevic@tenstorrent.com>
Date: Thu, 27 Jun 2024 14:16:24 +0000
Subject: [PATCH] Update gitlab action to build for wormhole_b0

---
 .github/workflows/build-artifacts.yml       | 15 +++++++++++++--
 .github/workflows/post-commit-workflow.yml  | 13 -------------
 .github/workflows/pull-request-workflow.yml | 13 -------------
 3 files changed, 13 insertions(+), 28 deletions(-)
 delete mode 100644 .github/workflows/post-commit-workflow.yml
 delete mode 100644 .github/workflows/pull-request-workflow.yml

diff --git a/.github/workflows/build-artifacts.yml b/.github/workflows/build-artifacts.yml
index 5e52e7e8..e272c417 100644
--- a/.github/workflows/build-artifacts.yml
+++ b/.github/workflows/build-artifacts.yml
@@ -3,15 +3,26 @@ name: Build artifacts
 on:
   workflow_dispatch:
   workflow_call:
+  push:
+    branches:
+      - main
+  pull_request:
+    branches:
+      - main
 
 env:
   PYTHON_VERSION: "python3.10"
 
 jobs:
   build-artifacts:
+    
     strategy:
       matrix:
-        arch: ["grayskull"]
+        include:
+          - arch: grayskull
+            env_script: env_for_silicon.sh
+          - arch: wormhole_b0
+            env_script: env_for_wormhole_b0.sh
     runs-on: ubuntu-22.04
     steps:
       - uses: actions/checkout@v4
@@ -21,4 +32,4 @@ jobs:
       - name: Update submodule
         run: git submodule update --init --recursive
       - name: Build for ${{ matrix.arch }}
-        run: source env_for_silicon.sh
\ No newline at end of file
+        run: source ${{ matrix.env_script }}
diff --git a/.github/workflows/post-commit-workflow.yml b/.github/workflows/post-commit-workflow.yml
deleted file mode 100644
index ceb7d58d..00000000
--- a/.github/workflows/post-commit-workflow.yml
+++ /dev/null
@@ -1,13 +0,0 @@
-name: Post commit workflow
-
-on:
-  workflow_dispatch:
-  workflow_call:
-  push:
-    branches:
-      - main
-
-jobs:
-  build-artifacts:
-    uses: ./.github/workflows/build-artifacts.yml
-    secrets: inherit
\ No newline at end of file
diff --git a/.github/workflows/pull-request-workflow.yml b/.github/workflows/pull-request-workflow.yml
deleted file mode 100644
index c5fbe795..00000000
--- a/.github/workflows/pull-request-workflow.yml
+++ /dev/null
@@ -1,13 +0,0 @@
-name: Pull request workflow
-
-on:
-  workflow_dispatch:
-  workflow_call:
-  pull_request:
-    branches:
-      - main
-
-jobs:
-  build-artifacts:
-    uses: ./.github/workflows/build-artifacts.yml
-    secrets: inherit
\ No newline at end of file