From 7c20bf6a4ba592e80a16ee70ec95bf2ae1f268a3 Mon Sep 17 00:00:00 2001
From: Surabhi Jain <surabhi.jain@intel.com>
Date: Fri, 7 Sep 2018 17:23:32 -0500
Subject: [PATCH] test: Add bcast, reduce tests for newly added CVARS

Run a few bcast and reduce tests by varying the CVARS for multiple
buffer sizes and type, radix of trees.
---
 test/mpi/coll/test_coll_algos.sh | 58 ++++++++++++++++++++++++++++++++
 1 file changed, 58 insertions(+)

diff --git a/test/mpi/coll/test_coll_algos.sh b/test/mpi/coll/test_coll_algos.sh
index edd85148441..26a91c115dd 100755
--- a/test/mpi/coll/test_coll_algos.sh
+++ b/test/mpi/coll/test_coll_algos.sh
@@ -443,4 +443,62 @@ for algo_name in ${algo_names}; do
     coll_algo_tests+="neighb_alltoallv 4 mpiversion=3.0 ${env}${nl}"
 done
 
+########## Add tests for intra-node bcast algorithms ############
+
+#use release gather based intra-node bcast
+testing_env="env=MPIR_CVAR_BCAST_POSIX_INTRA_ALGORITHM=release_gather "
+
+testing_env+="env=MPIR_CVAR_COLL_SHM_LIMIT_PER_NODE=131072 " #128MB
+buffer_sizes="16384 32768"
+num_cells="2 4"
+tree_types="knomial_1 knomial_2"
+kvalues="8 64"
+
+for buf_size in ${buffer_sizes}; do
+    for num_cell in ${num_cells}; do
+        for kval in ${kvalues}; do
+            for tree_type in ${tree_types}; do
+                #set the environment
+                env="${testing_env} env=MPIR_CVAR_BCAST_INTRANODE_BUFFER_TOTAL_SIZE=${buf_size} "
+                env+="env=MPIR_CVAR_BCAST_INTRANODE_NUM_CELLS=${num_cell} "
+                env+="env=MPIR_CVAR_BCAST_INTRANODE_TREE_KVAL=${kval} "
+                env+="env=MPIR_CVAR_BCAST_INTRANODE_TREE_TYPE=${tree_type} "
+
+                coll_algo_tests+="bcasttest 10 ${env}${nl}"
+                coll_algo_tests+="bcastzerotype 5 ${env}${nl}"
+            done
+        done
+    done
+done
+
+########## Add tests for intra-node reduce algorithms ############
+
+#use release gather based intra-node reduce
+testing_env="env=MPIR_CVAR_REDUCE_POSIX_INTRA_ALGORITHM=release_gather "
+
+testing_env+="env=MPIR_CVAR_COLL_SHM_LIMIT_PER_NODE=131072 " #128MB
+buffer_sizes="16384 32768"
+num_cells="2 4"
+tree_types="knomial_1 knomial_2"
+kvalues="4 8"
+
+for buf_size in ${buffer_sizes}; do
+    for num_cell in ${num_cells}; do
+        for kval in ${kvalues}; do
+            for tree_type in ${tree_types}; do
+                #set the environment
+                env="${testing_env} env=MPIR_CVAR_REDUCE_INTRANODE_BUFFER_TOTAL_SIZE=${buf_size} "
+                env+="env=MPIR_CVAR_REDUCE_INTRANODE_NUM_CELLS=${num_cell} "
+                env+="env=MPIR_CVAR_REDUCE_INTRANODE_TREE_KVAL=${kval} "
+                env+="env=MPIR_CVAR_REDUCE_INTRANODE_TREE_TYPE=${tree_type} "
+
+                coll_algo_tests+="reduce 5 ${env}${nl}"
+                coll_algo_tests+="reduce 10 ${env}${nl}"
+                coll_algo_tests+="red3 10 ${env}${nl}"
+                coll_algo_tests+="red4 10 ${env}${nl}"
+            done
+        done
+    done
+done
+
 export coll_algo_tests