Add 'projects/rocshmem/' from commit '0496586829058af5cfd7f23acda2a6d0040da584'

git-subtree-dir: projects/rocshmem
git-subtree-mainline: 5fd976da70
git-subtree-split: 0496586829

projects/rocshmem/scripts/unit_tests/driver.sh

@@ -0,0 +1,99 @@
+###############################################################################
+# Copyright (c) Advanced Micro Devices, Inc. All rights reserved.
+#
+# SPDX-License-Identifier: MIT
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to
+# deal in the Software without restriction, including without limitation the
+# rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
+# sell copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in
+# all copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
+# FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
+# IN THE SOFTWARE.
+###############################################################################
+
+#!/bin/bash
+
+# Function to display help information
+function display_help {
+    echo "Usage:"
+    echo "  $0 binary_name all                     # Runs all standard tests"
+    echo "  $0 binary_name custom <ranks> <filter> # Runs custom test configuration"
+    echo
+    echo "Arguments:"
+    echo "  binary_name: Name of the binary to run."
+    echo "  all: Executes predefined test configurations."
+    echo "  custom: Executes a test with custom MPI ranks and GTest filter."
+    echo "  ranks: Number of MPI ranks (required for custom mode)."
+    echo "  filter: GTest filter string (required for custom mode)."
+    echo
+}
+
+# Validate number of arguments for each mode
+if [[ "$#" -lt 2 ]] ||
+   { [[ "$2" == "all" ]] && [[ "$#" -ne 2 ]]; } ||
+   { [[ "$2" == "custom" ]] && [[ "$#" -ne 4 ]]; }; then
+    display_help
+    exit 1
+fi
+
+driver_return_status=0
+binary_name=$1
+mode=$2
+timestamp=$(date "+%Y-%m-%d-%H:%M:%S")
+log_file="unit_tests_${timestamp}.log"
+mpi_timeout=$((20 * 60)) # 20 minutes in seconds
+
+# Function to execute mpirun command
+function run_mpirun {
+    local np=$1
+    local gtest_filter=$2
+    cmd_str="mpirun -np $np --timeout $mpi_timeout $binary_name --gtest_filter=$gtest_filter >> $log_file 2>&1"
+    echo $cmd_str
+    eval $cmd_str
+
+    # Test if mpirun failed
+    if [ $? -ne 0 ]
+    then
+        echo "FAILED: $cmd_str" >&2
+        cat $log_file
+        driver_return_status=1
+    fi
+}
+
+# Processing modes
+case $mode in
+    all)
+        test_with_two_pes="IPCImplSimpleCoarseTestFixture/*:IPCImplSimpleFineTestFixture/*:IPCImplTiledFineTestFixture/*:DegenerateTiledFine.*"
+        run_mpirun 4 "-$test_with_two_pes"
+        #run_mpirun 2 "$test_with_two_pes"
+        ;;
+    custom)
+        # Check if ranks is a positive integer
+        if [[ "$3" -le 1 ]]; then
+            echo "Error: 'ranks' must be a positive integer."
+            display_help
+            exit 1
+        fi
+        run_mpirun $3 $4
+        ;;
+    *)
+        echo "Error: Invalid mode '$mode'." | tee -a "$log_file"
+        display_help
+        exit 1
+        ;;
+esac
+
+echo "Tests Completed"
+echo "log file: '$log_file'"
+exit $driver_return_status