Skip to content

Commit

Permalink
Add new option "mpirun_options" to config file
Browse files Browse the repository at this point in the history
With the addition of "mpirun_options" to the config file it is easy to
run specific mapping and pinning configurations.
An example for mapping one process per node is provided.

Signed-off-by: Christoph Niethammer <niethammer@hlrs.de>
  • Loading branch information
cniethammer committed May 19, 2020
1 parent 58fe5df commit 20aaeeb
Show file tree
Hide file tree
Showing 3 changed files with 11 additions and 0 deletions.
2 changes: 2 additions & 0 deletions coltune_script.py
Original file line number Diff line number Diff line change
Expand Up @@ -30,6 +30,7 @@ def main():
max_num_node = config.getInt("max_num_node")
num_core_per_node = config.getInt("number_of_cores_per_node")
num_run = config.getInt("number_of_runs_per_test")
mpirun_options = config.getStr("mpirun_options")

job_directory = dir_path+"/collective_jobs"
for collective in collective_list:
Expand Down Expand Up @@ -82,6 +83,7 @@ def main():
else:
prg_name = omb_path+"/osu_"+collective
cmd = "mpirun --np %d " % (num_rank)
cmd += "%s " % (mpirun_options)
cmd += "--mca coll_tuned_use_dynamic_rules 1 --mca coll_tuned_"+collective+"_algorithm "+str(alg)
cmd += " " + prg_name
cmd += " >& " + dir_path+"/output/"+collective + "/" + str(alg) + "_" + str(num_rank) + "ranks" + "_run" + str(run_id) + ".out"
Expand Down
1 change: 1 addition & 0 deletions example/config
Original file line number Diff line number Diff line change
Expand Up @@ -5,3 +5,4 @@ number_of_ranks : 2 4 8 16 32 64
max_num_node : 2
number_of_cores_per_node : 36
number_of_runs_per_test : 3
mpirun_options :
8 changes: 8 additions & 0 deletions example/config-one-proc-per-node
Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
collectives : allgather allgatherv allreduce alltoall alltoallv barrier bcast gather reduce reduce_scatter_block reduce_scatter scatter
omb_collective_directory : /path/to/install/libexec/osu-micro-benchmarks/mpi/collective/
imb_binary : /path/to/install/IMB-MPI1
number_of_ranks : 2 3 4 5 6 7 8
max_num_node : 8
number_of_cores_per_node : 1
number_of_runs_per_test : 3
mpirun_options : --map-by ppr:1:node

0 comments on commit 20aaeeb

Please sign in to comment.