From adc762ec139cf8a504bb115ff9c5f95efbccbd5b Mon Sep 17 00:00:00 2001
From: Garrett Mills <g582m300@ku.edu>
Date: Wed, 1 May 2019 16:05:34 -0500
Subject: [PATCH] fix sample means example

---
 demos/sample_means/generate_plots.R               | 3 ++-
 demos/sample_means/mpi_generate_plots.sh          | 5 +++++
 demos/sample_means/plots/.gitkeep                 | 0
 demos/sample_means/slurm_submit_generate_plots.sh | 5 ++---
 4 files changed, 9 insertions(+), 4 deletions(-)
 create mode 100644 demos/sample_means/mpi_generate_plots.sh
 create mode 100644 demos/sample_means/plots/.gitkeep

diff --git a/demos/sample_means/generate_plots.R b/demos/sample_means/generate_plots.R
index 522ec04..ec8de73 100644
--- a/demos/sample_means/generate_plots.R
+++ b/demos/sample_means/generate_plots.R
@@ -7,6 +7,7 @@ nodes = mpi.universe.size()
 mpi.spawn.Rslaves(nslaves=nodes)
 
 generate_plot <- function(filename){
+	print(mpi.comm.rank())
 	samples = rep(NA, 100000)
 	for ( i in 1:100000 ){ samples[i] = mean(rexp(40, 0.2)) }
 	
@@ -20,7 +21,7 @@ generate_plot <- function(filename){
 
 samples = 1:number_of_plots
 
-mpi.apply(samples, generate_plot)
+mpi.applyLB(samples, generate_plot)
 
 mpi.close.Rslaves()
 mpi.quit()
diff --git a/demos/sample_means/mpi_generate_plots.sh b/demos/sample_means/mpi_generate_plots.sh
new file mode 100644
index 0000000..2700e4b
--- /dev/null
+++ b/demos/sample_means/mpi_generate_plots.sh
@@ -0,0 +1,5 @@
+#!/bin/bash
+
+mkdir plots/
+
+mpiexec --allow-run-as-root -n 1 --hostfile /kube-openmpi/generated/hostfile R --vanilla -f generate_plots.R > mpi.out
diff --git a/demos/sample_means/plots/.gitkeep b/demos/sample_means/plots/.gitkeep
new file mode 100644
index 0000000..e69de29
diff --git a/demos/sample_means/slurm_submit_generate_plots.sh b/demos/sample_means/slurm_submit_generate_plots.sh
index 04207d8..2f21a94 100644
--- a/demos/sample_means/slurm_submit_generate_plots.sh
+++ b/demos/sample_means/slurm_submit_generate_plots.sh
@@ -1,9 +1,8 @@
 #!/bin/bash
-#SBATCH --nodes=1
-#SBATCH --ntasks-per-node=1
+#SLURM --ntasks=5
 
 cd $SLURM_SUBMIT_DIR
 
 mkdir plots/
 
-R --vanilla -f generate_plots.R --args "plot$SLURM_ARRAY_TASK_ID"
+mpiexec -n 1 R --vanilla -f generate_plots.R