[ci] Enabling test labels for workflow dispatch (#1805)

## Motivation

When working with MIOpen team to trigger specific tests, I noticed that
the `test_labels` functionality for workflow dispatch was not working

## Technical Details

with this change, developers can specify which full tests to run instead
of having to wait on specific tests

## Test Plan

Verifying with Github Actions

## Test Result

Works here:
https://github.com/ROCm/TheRock/actions/runs/18545752517/job/52863296833

## Submission Checklist

- [x] Look over the contributing guidelines at
https://github.com/ROCm/ROCm/blob/develop/CONTRIBUTING.md#pull-requests.

Author: geomin12

.github/workflows/setup.yml

@@ -58,7 +58,9 @@ jobs:
         id: configure
         env:
           INPUT_LINUX_AMDGPU_FAMILIES: ${{ github.event.inputs.linux_amdgpu_families }}
+          LINUX_TEST_LABELS: ${{ github.event.inputs.linux_test_labels }}
           LINUX_USE_PREBUILT_ARTIFACTS: ${{ github.event.inputs.linux_use_prebuilt_artifacts }}
           INPUT_WINDOWS_AMDGPU_FAMILIES: ${{ github.event.inputs.windows_amdgpu_families }}
+          WINDOWS_TEST_LABELS: ${{ github.event.inputs.windows_test_labels }}
           WINDOWS_USE_PREBUILT_ARTIFACTS: ${{ github.event.inputs.windows_use_prebuilt_artifacts }}
         run: ./build_tools/github_actions/configure_ci.py

build_tools/github_actions/configure_ci.py

@@ -11,8 +11,10 @@
   * GITHUB_OUTPUT        : path to write workflow output variables.
   * GITHUB_STEP_SUMMARY  : path to write workflow summary output.
   * INPUT_LINUX_AMDGPU_FAMILIES (optional): Comma-separated string of Linux AMD GPU families
+  * LINUX_TEST_LABELS (optional): Comma-separated list of test labels to test
   * LINUX_USE_PREBUILT_ARTIFACTS (optional): If enabled, CI will only run Linux tests
   * INPUT_WINDOWS_AMDGPU_FAMILIES (optional): Comma-separated string of Windows AMD GPU families
+  * WINDOWS_TEST_LABELS (optional): Comma-separated list of test labels to test
   * WINDOWS_USE_PREBUILT_ARTIFACTS (optional): If enabled, CI will only run Windows tests
   * BRANCH_NAME (optional): The branch name
 
@@ -280,6 +282,25 @@ def matrix_generator(
             filter_known_names(requested_target_names, "target")
         )
 
+        # If any workflow dispatch test labels are specified, we run full tests for those specific tests
+        workflow_dispatch_test_labels_str = (
+            base_args.get("workflow_dispatch_linux_test_labels", "")
+            if platform == "linux"
+            else base_args.get("workflow_dispatch_windows_test_labels", "")
+        )
+        # (ex: "test:rocprim, test:hipcub" -> ["test:rocprim", "test:hipcub"])
+        workflow_dispatch_test_labels = [
+            test_label.strip()
+            for test_label in workflow_dispatch_test_labels_str.split(",")
+        ]
+
+        requested_test_names = []
+        for label in workflow_dispatch_test_labels:
+            if "test:" in label:
+                _, test_name = label.split(":")
+                requested_test_names.append(test_name)
+        selected_test_names.extend(filter_known_names(requested_test_names, "test"))
+
     if is_pull_request:
         print(f"[PULL_REQUEST] Generating build matrix with {str(base_args)}")
 
@@ -458,5 +479,11 @@ def main(base_args, linux_families, windows_families):
     base_args["windows_use_prebuilt_artifacts"] = (
         os.environ.get("WINDOWS_USE_PREBUILT_ARTIFACTS") == "true"
     )
+    base_args["workflow_dispatch_linux_test_labels"] = os.getenv(
+        "LINUX_TEST_LABELS", ""
+    )
+    base_args["workflow_dispatch_windows_test_labels"] = os.getenv(
+        "WINDOWS_TEST_LABELS", ""
+    )
 
     main(base_args, linux_families, windows_families)

build_tools/github_actions/tests/configure_ci_test.py

@@ -89,7 +89,10 @@ def test_valid_linux_workflow_dispatch_matrix_generator(self):
             is_workflow_dispatch=True,
             is_push=False,
             is_schedule=False,
-            base_args={},
+            base_args={
+                "workflow_dispatch_linux_test_labels": "",
+                "workflow_dispatch_windows_test_labels": "",
+            },
             families=build_families,
             platform="linux",
         )