[rocm-libraries] ROCm/rocm-libraries#7215 (commit 955d3a8)

Add test filter standardization to rocthrust

## Motivation

Add test filter standardization to rocthrust.
The goal is to categorize tests at the component level, enabling TheRock
CI to invoke a common test_runner with specified categories to run the
appropriate tests.

More details available at
[test_filtering.md](https://github.com/ROCm/TheRock/blob/main/docs/development/test_filtering.md).

## Technical Details

These following are the changes made in rocm-libs:
Add test_categories.yaml in rocthrust.
Update CMakeLists.txt to call the parser script
parse_ctest_categories.py to add test category labels.
This adds additional ctest entries with with corresponding category
labels which can be used as in test plan.

## Test Plan

Confirm that ctest commands and tests work good.
ctest --print-labels to see the labels added
ctest -L quick to run the quick tests

## Test Result

Test run:
https://github.com/ROCm/rocm-libraries/actions/runs/25927754134/job/76259007137?pr=7215

## Submission Checklist

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

Author: dileepr1

CMakeLists.txt

@@ -46,6 +46,10 @@ set(CMAKE_HIP_STANDARD 17)
 set(CMAKE_HIP_STANDARD_REQUIRED ON)
 set(CMAKE_HIP_EXTENSIONS OFF)
 
+if(NOT DEFINED ROCM_LIBRARIES_ROOT)
+  set(ROCM_LIBRARIES_ROOT "${CMAKE_CURRENT_SOURCE_DIR}/../..")
+endif()
+
 include(CheckLanguage)
 include(CMakeDependentOption)
 

test/CMakeLists.txt

@@ -96,6 +96,10 @@ function(add_rocthrust_test TEST)
         PROPERTIES
             RUNTIME_OUTPUT_DIRECTORY "${CMAKE_BINARY_DIR}/test/"
     )
+
+    # Track test target globally for category labels
+    set_property(GLOBAL APPEND PROPERTY ROCTHRUST_TEST_TARGETS ${TEST_TARGET})
+
     if(AMDGPU_TEST_TARGETS)
         foreach(AMDGPU_TARGET IN LISTS AMDGPU_TEST_TARGETS)
             add_test("${AMDGPU_TARGET}-${TEST_TARGET}" ${TEST_TARGET})
@@ -340,6 +344,30 @@ if(BUILD_HIPSTDPAR_TEST)
     endif()
 endif()
 
+# ****************************************************************************
+# Test Filter Standardization
+# ****************************************************************************
+
+if(EXISTS "${ROCM_LIBRARIES_ROOT}/shared/ctest/TestCategories.cmake")
+    include("${ROCM_LIBRARIES_ROOT}/shared/ctest/TestCategories.cmake")
+
+    message(STATUS "Applying test categories for rocthrust")
+    get_property(_rt_test_targets GLOBAL PROPERTY ROCTHRUST_TEST_TARGETS)
+    foreach(_rt_t IN LISTS _rt_test_targets)
+        get_target_property(_rt_dir ${_rt_t} RUNTIME_OUTPUT_DIRECTORY)
+        apply_test_category_labels(${_rt_t}
+            "${CMAKE_CURRENT_SOURCE_DIR}/test_categories.yaml"
+            "${_rt_dir}" "${INSTALL_TEST_FILE}")
+    endforeach()
+else()
+    message(
+        STATUS
+        "Shared CTest categories not found at "
+        "${ROCM_LIBRARIES_ROOT}/shared/ctest/TestCategories.cmake - "
+        "skipping rocthrust test category labels"
+    )
+endif()
+
 rocm_install(
     FILES "${INSTALL_TEST_FILE}"
     DESTINATION "${CMAKE_INSTALL_BINDIR}/${PROJECT_NAME}"

test/test_categories.yaml

@@ -0,0 +1,110 @@
+# rocThrust test categories
+#
+# This file is the single source of truth for the gtest filter that drives the
+# `quick` CTest tier in CI. The patterns below mirror the QUICK_TESTS list in
+# TheRock's `build_tools/github_actions/test_executable_scripts/test_rocthrust.py`
+# so that running `ctest -L quick` from the build / install tree produces the
+# same selection that TheRock CI runs via the GTEST_FILTER environment
+# variable.
+#
+# Each entry under `quick.test_patterns` is a gtest filter pattern. The shared
+# parser (rocm-libraries/shared/ctest/parse_test_categories.py) joins them with
+# ':' and registers a `<binary>_quick_suite` CTest entry per rocThrust test
+# binary, invoking that binary with `--gtest_filter=<joined patterns>`.
+#
+test_categories:
+  quick:
+    description: "Fast sanity checks - mirrors TheRock test_rocthrust.py QUICK_TESTS"
+    test_patterns:
+      - "AllocatorTests.*"
+      - "AsyncExclusiveScan*"
+      - "AsyncInclusiveScan*"
+      - "AsyncReduce*"
+      - "AsyncSort*"
+      - "AsyncTransform*"
+      - "AsyncTriviallyRelocatableElements*"
+      - "ConstantIteratorTests.*"
+      - "Copy*"
+      - "CopyN*"
+      - "Count*"
+      - "CountingIteratorTests.*"
+      - "Dereference*"
+      - "DeviceDelete*"
+      - "DevicePathSimpleTest"
+      - "DevicePtrTests.*"
+      - "DeviceReferenceTests.*"
+      - "DiscardIteratorTests.*"
+      - "EqualTests.*"
+      - "Fill*"
+      - "Find*"
+      - "ForEach*"
+      - "Gather*"
+      - "Generate*"
+      - "InnerProduct*"
+      - "IsPartitioned*"
+      - "IsSorted*"
+      - "IsSortedUntil*"
+      - "MemoryTests.*"
+      - "Merge*"
+      - "MergeByKey*"
+      - "Mr*Tests.*"
+      - "Partition*"
+      - "PartitionPoint*"
+      - "PermutationIteratorTests.*"
+      - "RandomTests.*"
+      - "Reduce*"
+      - "ReduceByKey*"
+      - "Remove*"
+      - "RemoveIf*"
+      - "Replace*"
+      - "ReverseIterator*"
+      - "Scan*"
+      - "ScanByKey*"
+      - "Scatter*"
+      - "Sequence*"
+      - "SetDifference*"
+      - "SetIntersection*"
+      - "SetSymmetricDifference*"
+      - "Shuffle*"
+      - "Sort*"
+      - "StableSort*"
+      - "StableSortByKey*"
+      - "Tabulate*"
+      - "TestBijectionLength"
+      - "TestHipThrustCopy.DeviceToDevice"
+      - "Transform*"
+      - "TransformIteratorTests.*"
+      - "TransformReduce*"
+      - "TransformScan*"
+      - "UninitializedCopy*"
+      - "UninitializedFill*"
+      - "Unique*"
+      - "Vector*"
+      - "VectorAllocatorTests.*"
+      - "ZipIterator*"
+    exclude:
+    labels:
+      - "quick"
+
+  standard:
+    description: "Full gtest run (filter '*') - shared across PR / nightly / weekly tiers"
+    test_patterns:
+      - "*"
+    exclude:
+    labels:
+      - "standard"
+      - "comprehensive"
+      - "full"
+
+# No GPU-specific exclusions are defined yet for rocThrust. Add entries here if a
+# specific gfx target needs to skip particular suites (see rocPRIM's
+# test_categories.yaml for a worked example). The empty mapping is required so
+# the shared parser does not treat the section as `None`.
+exclude_gpu: {}
+
+execution_settings:
+  default_timeout: 300
+  timeout_multiplier: 1 # Multiplier for all timeouts (1, 1.5, 1.75, 2, etc.)
+  category_timeouts:
+    quick: 300          # 5 minutes
+    standard: 14400     # 4 hours (longest tier the standard entry serves)