From e5032c73e6bfd9bd99b27c0564ec27d724f315bd Mon Sep 17 00:00:00 2001
From: Achim Gsell <achim.gsell@psi.ch>
Date: Thu, 1 Feb 2018 11:25:18 +0100
Subject: [PATCH] Compiler/openmpi/build: - build with CUDA support on Linux

---
 Compiler/openmpi/build | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/Compiler/openmpi/build b/Compiler/openmpi/build
index c002bcf..6ead2e5 100755
--- a/Compiler/openmpi/build
+++ b/Compiler/openmpi/build
@@ -3,6 +3,13 @@
 SOURCE_URL="http://www.open-mpi.org/software/ompi/v${V_MAJOR}.${V_MINOR}/downloads/openmpi-$V.tar.bz2"
 
 pbuild::configure() {
+	declare -a config_args=()
+	case "$OS" in
+		Linux )
+			config_args+=( "--with-cuda=${CUDA_HOME}" )
+			;;
+	esac
+
 	"${MODULE_SRCDIR}"/configure \
 		--prefix="${PREFIX}" \
 		--enable-mpi-cxx \
@@ -14,6 +21,7 @@ pbuild::configure() {
 		--enable-smp-locks \
 		--with-sge=yes \
 		--with-slurm=yes \
+		"${config_args[@]}" \
 		|| exit 1
 }