Merge branch 'ROCm:develop' into functional_tests/puts_gets

[ROCm/rocshmem commit: e981f61693]
2024-10-14 10:27:54 -05:00
@@ -0,0 +1,31 @@
+#!/bin/bash
+# Copyright (c) 2024 Advanced Micro Devices, Inc. All rights reserved.
+
+if [ -z $1 ]
+then
+  install_path=~/rocshmem
+else
+  install_path=$1
+fi
+
+src_path=$(dirname "$(realpath $0)")/../../
+
+cmake \
+    -DCMAKE_BUILD_TYPE=Release \
+    -DCMAKE_INSTALL_PREFIX=$install_path \
+    -DCMAKE_VERBOSE_MAKEFILE=OFF \
+    -DDEBUG=OFF \
+    -DPROFILE=OFF \
+    -DUSE_GPU_IB=OFF \
+    -DUSE_RO=OFF \
+    -DUSE_DC=OFF \
+    -DUSE_IPC=ON \
+    -DUSE_COHERENT_HEAP=ON \
+    -DUSE_THREADS=OFF \
+    -DUSE_WF_COAL=OFF \
+    -DUSE_SINGLE_NODE=ON \
+    -DUSE_HOST_SIDE_HDP_FLUSH=OFF \
+    -DUSE_COOPERATIVE_GROUPS=ON \
+    $src_path
+cmake --build . --parallel 8
+cmake --install .
@@ -20,7 +20,6 @@ cmake \
    -DUSE_DC=OFF \
    -DUSE_IPC=OFF \
    -DUSE_COHERENT_HEAP=OFF \
-    -DUSE_CACHED_HEAP=OFF \
    -DUSE_THREADS=OFF \
    -DUSE_WF_COAL=OFF \
    $src_path
@@ -20,12 +20,10 @@ cmake \
    -DUSE_DC=OFF \
    -DUSE_IPC=ON \
    -DUSE_COHERENT_HEAP=OFF \
-    -DUSE_CACHED_HEAP=OFF \
    -DUSE_THREADS=OFF \
    -DUSE_WF_COAL=OFF \
    -DUSE_SINGLE_NODE=ON \
    -DUSE_HOST_SIDE_HDP_FLUSH=ON\
-    -DROCM_PATH="/opt/rocm-5.4.2/"\
    $src_path
 cmake --build . --parallel 8
 cmake --install .
@@ -20,7 +20,6 @@ cmake \
    -DUSE_DC=OFF \
    -DUSE_IPC=ON \
    -DUSE_COHERENT_HEAP=OFF \
-    -DUSE_CACHED_HEAP=OFF \
    -DUSE_THREADS=OFF \
    -DUSE_WF_COAL=OFF \
    -DUSE_SINGLE_NODE=ON \
@@ -0,0 +1,29 @@
+#!/bin/bash
+# Copyright (c) 2024 Advanced Micro Devices, Inc. All rights reserved.
+
+if [ -z $1 ]
+then
+  install_path=~/rocshmem
+else
+  install_path=$1
+fi
+
+src_path=$(dirname "$(realpath $0)")/../../
+
+cmake \
+    -DCMAKE_BUILD_TYPE=Release \
+    -DCMAKE_INSTALL_PREFIX=$install_path \
+    -DCMAKE_VERBOSE_MAKEFILE=OFF \
+    -DDEBUG=OFF \
+    -DPROFILE=OFF \
+    -DUSE_GPU_IB=OFF \
+    -DUSE_DC=OFF \
+    -DUSE_IPC=ON \
+    -DUSE_COHERENT_HEAP=ON \
+    -DUSE_THREADS=OFF \
+    -DUSE_WF_COAL=OFF \
+    -DUSE_SINGLE_NODE=ON \
+    -DUSE_HOST_SIDE_HDP_FLUSH=OFF\
+    $src_path
+cmake --build . --parallel 8
+cmake --install .
@@ -22,7 +22,7 @@ cmake \
    -DUSE_IPC=OFF \
    -DUSE_THREADS=ON \
    -DUSE_WF_COAL=OFF \
-    -DUSE_COHERENT_HEAP=ON \
+    -DUSE_COHERENT_HEAP=OFF \
    $src_path
 cmake --build . --parallel 8
 cmake --install .
@@ -21,7 +21,7 @@ cmake \
    -DUSE_IPC=OFF \
    -DUSE_THREADS=ON \
    -DUSE_WF_COAL=OFF \
-    -DUSE_COHERENT_HEAP=ON \
+    -DUSE_COHERENT_HEAP=OFF \
    $src_path
 cmake --build . --parallel 8
 cmake --install .
@@ -121,9 +121,9 @@ case $2 in
        echo "amoinc_n2_w1_z1"
        ROC_SHMEM_MAX_NUM_CONTEXTS=1 mpirun -np 2 $1 -w 1 -z 1 -a 11 > $3/amoinc_n2_w1_z1.log
        check amoinc_n2_w1_z1
-        echo "pingpong_n2_w1"
-        ROC_SHMEM_MAX_NUM_CONTEXTS=1 mpirun -np 2 $1 -w 1 -a 14 > $3/pingpong_n2_w1.log
-        check pingpong_n2_w1
+        # echo "pingpong_n2_w1"
+        # ROC_SHMEM_MAX_NUM_CONTEXTS=1 mpirun -np 2 $1 -w 1 -a 14 > $3/pingpong_n2_w1.log
+        # check pingpong_n2_w1
        echo "amoset_n2_w1_z1"
        ROC_SHMEM_MAX_NUM_CONTEXTS=1 mpirun -np 2 $1 -w 1 -z 1 -a 44 > $3/amoset_n2_w1_z1.log
        check amoset_n2_w1_z1
@@ -163,9 +163,9 @@ case $2 in
        echo "amoinc_n2_w8_z1"
        ROC_SHMEM_MAX_NUM_CONTEXTS=8 mpirun -np 2 $1 -w 8 -z 1 -a 11 > $3/amoinc_n2_w8_z1.log
        check amoinc_n2_w8_z1
-        echo "pingpong_n2_w1"
-        ROC_SHMEM_MAX_NUM_CONTEXTS=1 mpirun -np 2 $1 -w 1 -a 14 > $3/pingpong_n2_w1.log
-        check pingpong_n2_w1
+        # echo "pingpong_n2_w1"
+        # ROC_SHMEM_MAX_NUM_CONTEXTS=1 mpirun -np 2 $1 -w 1 -a 14 > $3/pingpong_n2_w1.log
+        # check pingpong_n2_w1
        echo "amoset_n2_w8_z1"
        ROC_SHMEM_MAX_NUM_CONTEXTS=8 mpirun -np 2 $1 -w 8 -z 1 -a 44 > $3/amoset_n2_w8_z1.log
        check amoset_n2_w8_z1
@@ -0,0 +1,62 @@
+#!/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
+
+binary_name=$1
+mode=$2
+timestamp=$(date "+%Y-%m-%d-%H:%M:%S")
+log_file="unit_tests_${timestamp}.log"
+
+# Function to execute mpirun command
+function run_mpirun {
+    local np=$1
+    local gtest_filter=$2
+    echo "mpirun -np $np $binary_name --gtest_filter='$gtest_filter'" | tee -a "$log_file"
+    mpirun -np "$np" "$binary_name" --gtest_filter="$gtest_filter" >> "$log_file" 2>&1
+}
+
+# Processing modes
+case $mode in
+    all)
+        run_mpirun 4 "-IPCImplSimpleCoarseTestFixture.*:IPCImplSimpleFineTestFixture.*"
+        run_mpirun 2 "IPCImplSimpleCoarseTestFixture.*:IPCImplSimpleFineTestFixture.*"
+        ;;
+    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'"