Ameya Keshava Mallya
12ab8df3bc
git-subtree-dir: projects/rocshmem git-subtree-mainline:5fd976da70git-subtree-split:0496586829
100 lines
3.4 KiB
Bash
Executable File
100 lines
3.4 KiB
Bash
Executable File
###############################################################################
|
|
# 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
|