diff --git a/script/install_MPI.sh b/script/install_MPI.sh index 583a881d84cb6a2265203e67730412b7a7ae5e5b..5a283b7a00b461be6814c9e0348d93804ab367b7 100755 --- a/script/install_MPI.sh +++ b/script/install_MPI.sh @@ -26,7 +26,13 @@ cd openmpi-3.1.1 # -./configure --with-cuda --prefix=$1/MPI --enable-mpi-fortran=yes CC=$3 CXX=$4 F77=$5 FC=$5 +if [ x"$6" == x"1" ]; then + echo "Installing MPI with GPU support" + ./configure --with-cuda --prefix=$1/MPI --enable-mpi-fortran=yes CC=$3 CXX=$4 F77=$5 FC=$5 +else + echo "Installing MPI without GPU support" + ./configure --prefix=$1/MPI --enable-mpi-fortran=yes CC=$3 CXX=$4 F77=$5 FC=$5 +fi make -j $2 make install