From 375530a1ac4101675d9348cb6de255f56cfd6a82 Mon Sep 17 00:00:00 2001
From: Michael Chiou <mchiou@tenstorrent.com>
Date: Wed, 15 Jan 2025 23:55:07 -0800
Subject: [PATCH] #0: Added custom tags for BH Post commit

---
 .github/workflows/blackhole-post-commit.yaml | 14 ++++++++++----
 1 file changed, 10 insertions(+), 4 deletions(-)

diff --git a/.github/workflows/blackhole-post-commit.yaml b/.github/workflows/blackhole-post-commit.yaml
index 2a3158a42a8..f7009d3805b 100644
--- a/.github/workflows/blackhole-post-commit.yaml
+++ b/.github/workflows/blackhole-post-commit.yaml
@@ -3,6 +3,12 @@ name: "Blackhole post-commit tests"
 on:
   workflow_call:
   workflow_dispatch:
+    inputs:
+      runner-label:
+          description: 'Optional: BH'
+          required: true
+          type: string
+          default: 'BH'
   schedule:
     - cron: "0 */2 * * *"
   # Pause this since not enough runners to support every commit to main
@@ -44,14 +50,14 @@ jobs:
     uses: ./.github/workflows/umd-unit-tests.yaml
     with:
       arch: blackhole
-      runner-label: BH
+      runner-label: ${{ inputs.runner-label }}
   sd-unit-tests:
     needs: build-artifact
     uses: ./.github/workflows/build-and-unit-tests.yaml
     secrets: inherit
     with:
       arch: blackhole
-      runner-label: BH
+      runner-label: ${{ inputs.runner-label }}
       timeout: 30
       os: "ubuntu-22.04"
   fd-unit-tests:
@@ -60,7 +66,7 @@ jobs:
     secrets: inherit
     with:
       arch: blackhole
-      runner-label: BH
+      runner-label: ${{ inputs.runner-label }}
       os: "ubuntu-22.04"
   # FD C++ Unit Tests
   cpp-unit-tests:
@@ -69,7 +75,7 @@ jobs:
     uses: ./.github/workflows/cpp-post-commit.yaml
     with:
       arch: blackhole
-      runner-label: BH
+      runner-label: ${{ inputs.runner-label }}
       timeout: 60
       os: "ubuntu-22.04"