Update CI to label-driven testing (#450)

* Initial commit

* Disabled lvl 1 flow when lvl 3 is used

* Made workflows mutually exclusive

* Updated with general label-driven level system

* Restore automatic testing on dev/release push

* Added inline comments, updated workflow names

* Added clarification on dispatch

* Streamlined cancel script updated build and test name

* Refactored dispatch to simplify and catch unlabel edge-case

* Updated to cull dispatch on unlabel when test already run

* Overhauled to leverage gh workflow concurrency, streamlined

* Removed run culling and streamlined

* Significantly simplified dispatch

* Updated copyright

Author: Micky774

.github/workflows/rocm-ci-dispatch.yml

@@ -0,0 +1,64 @@
+# Copyright (c) 2026, Advanced Micro Devices, Inc. All rights reserved.
+#
+# See LICENSE for license information.
+
+name: PR Automatic CI
+
+on:
+  pull_request:
+    branches:
+      - 'dev'
+      - 'release_v2.*_rocm'
+    types: [ opened, labeled, synchronize, reopened ]
+
+permissions:
+  contents: read
+
+jobs:
+  determine_level:
+    runs-on: ubuntu-latest
+    outputs:
+      test_level: ${{ steps.set_level.outputs.test_level }}
+    steps:
+      - name: Determine CI dispatch from labels
+        id: set_level
+        uses: actions/github-script@v7
+        with:
+          script: |
+            const parseLevelLabel = (labelName) => {
+              const m = (labelName || '').match(/^ci-level\s+(\d+)$/i);
+              return m ? Number(m[1]) : 0;
+            };
+
+            const labels = (context.payload.pull_request.labels || [])
+              .map(l => parseLevelLabel(l.name));
+            const action = context.payload.action;
+            const currentLevel = labels.length ? Math.max(...labels) : 0;
+
+            let dispatch = false;
+            if (action === 'labeled') {
+              const addedLevel = parseLevelLabel(context.payload.label?.name);
+              // Only dispatch when the newly added label is (or ties) the highest level.
+              dispatch = addedLevel > 0 && addedLevel === currentLevel;
+            } else {
+              // synchronize, reopened, opened — dispatch at whatever level is set.
+              dispatch = currentLevel > 0;
+            }
+
+            const level = dispatch ? String(currentLevel) : '';
+            core.info(`action=${action}, currentLevel=${currentLevel}, dispatch=${dispatch}`);
+            core.setOutput('test_level', level);
+
+  dispatch:
+    # Run this job if there is a valid level to test, which requires
+    # that any of the following are true:
+    # - A ci-level label higher than any existing ci-level label(s) was added
+    # - A commit was pushed with existing ci-level label(s)
+    # - The PR was reopened or opened with existing ci-level label(s)
+    if: ${{ needs.determine_level.outputs.test_level != '' }}
+    needs: determine_level
+    name: CI Level ${{ needs.determine_level.outputs.test_level }}
+    uses: ./.github/workflows/rocm-ci.yml
+    secrets: inherit
+    with:
+      test_level: ${{ needs.determine_level.outputs.test_level }}

.github/workflows/rocm-ci.yml

@@ -2,19 +2,30 @@
 #
 # See LICENSE for license information.
 
-name: TransformerEngine CI
+name: Build and Test Branch
+run-name: CI Level ${{ inputs.test_level || '1' }}
 
 on:
   push:
     branches:
       - 'dev'
-      - 'release_v1.*_rocm'
       - 'release_v2.*_rocm'
-  pull_request:
-    branches:
-      - 'dev'
-      - 'release_v1.**_rocm'
-      - 'release_v2.**_rocm'
+  workflow_call:
+    inputs:
+      test_level:
+        description: 'Test Level (1-3)'
+        required: false
+        default: '1'
+        type: string
+      docker_image_override:
+        description: 'Manual Docker Image (Leave empty to use config file value)'
+        required: false
+        type: string
+      test_config_from_source:
+        description: 'DEBUG: Use config.json from current source branch instead of dev'
+        required: false
+        default: false
+        type: boolean
   workflow_dispatch:
     inputs:
       test_level:
@@ -36,7 +47,7 @@ concurrency:
 
 jobs:
   build_and_test:
-    name: Build and Test on GPU (${{ matrix.runner }})
+    name: Build and Test on GPU (${{ matrix.runner }}) - Level ${{ inputs.test_level || '1' }}
     timeout-minutes: 720
     runs-on: ${{ matrix.runner }}
     strategy: