From bba9239924ff3257e4e499d25ce59e397ae2ee70 Mon Sep 17 00:00:00 2001
From: Vladimir Milosevic <vmilosevic@tenstorrent.com>
Date: Fri, 3 Jan 2025 14:40:31 +0000
Subject: [PATCH] Trigger tt-torch nd tt-forge-fe pipelines with mlir override

---
 .github/workflows/on-pr.yml | 28 ++++++++++++++++++----------
 1 file changed, 18 insertions(+), 10 deletions(-)

diff --git a/.github/workflows/on-pr.yml b/.github/workflows/on-pr.yml
index 478a9263f..3c4f9c6d3 100644
--- a/.github/workflows/on-pr.yml
+++ b/.github/workflows/on-pr.yml
@@ -4,6 +4,7 @@ on:
   workflow_dispatch:
   pull_request:
     branches: [ "main" ]
+  push: #debug
 
 concurrency:
   group: ${{ github.workflow }}-${{ github.ref }}
@@ -16,26 +17,33 @@ jobs:
   spdx:
     uses: ./.github/workflows/spdx.yml
     secrets: inherit
-  build-and-test:
-    uses: ./.github/workflows/build-and-test.yml
-    secrets: inherit
+  # build-and-test:
+  #   uses: ./.github/workflows/build-and-test.yml
+  #   secrets: inherit
 
   # When a PR runs on the uplift branch trigger the downstream checks
   downstream-checks:
     runs-on: ubuntu-latest
-    needs: build-and-test
-    if: github.event.pull_request.head.ref == 'uplift'
+    # needs: build-and-test
+    needs: pre-commit # debug
+    # if: github.event.pull_request.head.ref == 'uplift'
+    strategy:
+      matrix:
+        include:
+        - target-repo: 'tenstorrent/tt-forge-fe'
+          workflow-name: 'build-and-test.yml'
+        - target-repo: 'tenstorrent/tt-torch'
+          workflow-name: 'on-pr.yml'
     env:
-        GITHUB_TOKEN: ${{ secrets.GH_TOKEN }}
-        TARGET_REPO: tenstorrent/tt-forge-fe
-        WORKFLOW_NAME: build-and-test.yml
+      GITHUB_TOKEN: ${{ secrets.GH_TOKEN }}
+      TARGET_REPO: ${{ matrix.target-repo }}
+      WORKFLOW_NAME: ${{ matrix.workflow-name }}
     steps:
-      - name: Trigger ${{ env.TARGET_REPO }}
+      - name: Trigger ${{ env.TARGET_REPO }} tests
         shell: bash
         run: |
           gh workflow run ${{ env.WORKFLOW_NAME }} \
             --repo ${{ env.TARGET_REPO }} --ref main \
-            --field test_mark=push \
             --field mlir_override=${{ github.event.pull_request.head.sha }}
           gh run list --workflow=${{ env.WORKFLOW_NAME }} --repo ${{ env.TARGET_REPO }} --limit 1
           echo "Triggered ${{ env.TARGET_REPO }}"