scripts to reproduce experiments.

Project.toml

@@ -10,4 +10,5 @@ DelimitedFiles = "8bb1440f-4735-579b-a4ab-409b98df4dab"
 Libdl = "8f399da3-3557-5675-b5ff-fb832c97cbdb"
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
 MPI = "da04e1cc-30fd-572f-bb4f-1f8673147195"
+Plots = "91a5bcdd-55d7-5caf-9e0b-520d859cae80"
 Printf = "de0858da-6303-5e67-8744-51eddeeeb8d7"

figure5.sh

@@ -0,0 +1,31 @@
+#!/bin/bash
+#
+# This script describes how to reproduce the results of Figure 5.
+# This is just an example for iillustration purposes. Different platforms
+# (such as Summit cluster) may require different setups.
+#
+# For each run of admm_standalone.jl, it will generate iteration logs
+# and timing results. The relevant timing results for Figure 5 are printed
+# at the end of its run and will be the following:
+#
+#   Branch/iter = %.2f (millisecs)
+#
+# The above timing results were used for Figure 5.
+#
+# Prerequisite:
+#  - CUDA library files should be accessible before executing this script,
+#    e.g., module load cuda/10.2.89.
+#  - CUDA aware MPI should be available.
+
+export JULIA_CUDA_VERBOSE=1
+export JULIA_MPI_BINARY="system"
+
+DATA=("case2868rte" "case6515rte" "case9241pegase" "case13659pegase" "case19402_goc")
+PQ=(10 20 50 50 500)
+VA=(1000 2000 5000 5000 50000)
+ITER=(6000 15000 35000 45000 30000)
+
+for i in ${!DATA[@]}; do
+	julia --project ./src/admm_standalone.jl "./data/${DATA[$i]}" ${PQ[$i]} ${VA[$i]} ${ITER[$i]} true > output_gpu1_${DATA[$i]}.txt 2>&1
+done
+

figure6.sh

@@ -0,0 +1,35 @@
+#!/bin/bash
+#
+# This script describes how to reproduce the results of Figure 6.
+# This is just an example for iillustration purposes. Different platforms
+# (such as Summit cluster) may require different setups.
+#
+# For each run of launch_mpi.jl, it will generate iteration logs
+# and timing results. The relevant timing results for Figure 6 are printed
+# at the end of its run and will be the following:
+#
+#   (Br+MPI)/iter = %.2f (millisecs)
+#
+# We divide the above timing results by the timing results obtained when
+# we use a single GPU.
+#
+# Prerequisite:
+#  - CUDA library files should be accessible before executing this script,
+#    e.g., module load cuda/10.2.89.
+#  - CUDA aware MPI should be available.
+
+export JULIA_CUDA_VERBOSE=1
+export JULIA_MPI_BINARY="system"
+
+DATA=("case2868rte" "case6515rte" "case9241pegase" "case13659pegase" "case19402_goc")
+PQ=(10 20 50 50 500)
+VA=(1000 2000 5000 5000 50000)
+ITER=(6000 15000 35000 45000 30000)
+NGPU=(2 3 4 5 6)
+
+for j in ${!NGPU[@]}; do
+	for i in ${!DATA[@]}; do
+		mpirun -np ${j} julia --project ./src/launch_mpi.jl "./data/${DATA[$i]}" ${PQ[$i]} ${VA[$i]} ${ITER[$i]} true > output_gpu${j}_${DATA[$i]}.txt 2>&1
+        mv br_time_gpu.txt br_time_gpu${j}_${DATA[$i]}.txt
+	done
+done

figure7.sh

@@ -0,0 +1,21 @@
+#!/bin/bash
+#
+# This script describes how to reproduce the results of Figure 7.
+# This is just an example for iillustration purposes. Different platforms
+# (such as Summit cluster) may require different setups.
+#
+# We need br_time_13659pegase.txt file which is obtained when we run 
+# with 6 GPUs over 13659pegase example. The file can be obtained by 
+# running figure6.sh.
+
+function usage() {
+    echo "Usage: ./figure7.sh case"
+    echo "  case: the case file containing branch computation time of each GPU"
+}
+
+if [[ $# != 1 ]]; then
+    usage
+    exit
+fi
+
+julia --project ./src/heatmap.jl $1

figure8.sh

@@ -0,0 +1,31 @@
+#!/bin/bash
+#
+# This script describes how to reproduce the results of Figure 8.
+# This is just an example for iillustration purposes. Different platforms
+# (such as Summit cluster) may require different setups.
+#
+# For each run of launch_mpi.jl, it will generate iteration logs
+# and timing results. The relevant timing results for Figure 8 are printed
+# at the end of its run and will be the following:
+#
+#   (Br+MPI)/iter = %.2f (millisecs)
+#
+# We use these numbers for the timings of 40 CPU cores and use the timings
+# from Figure 6 for 6 GPUs.
+#
+# Prerequisite:
+#  - CUDA library files should be accessible before executing this script,
+#    e.g., module load cuda/10.2.89.
+#  - CUDA aware MPI should be available.
+
+export JULIA_CUDA_VERBOSE=1
+export JULIA_MPI_BINARY="system"
+
+DATA=("case2868rte" "case6515rte" "case9241pegase" "case13659pegase" "case19402_goc")
+PQ=(10 20 50 50 500)
+VA=(1000 2000 5000 5000 50000)
+ITER=(6000 15000 35000 45000 30000)
+
+for i in ${!DATA[@]}; do
+	mpirun -np 40 julia --project ./src/launch_mpi.jl "./data/${DATA[$i]}" ${PQ[$i]} ${VA[$i]} ${ITER[$i]} false > output_cpu40_${DATA[$i]}.txt 2>&1
+done