Add cuda variant for mvapich2. (#4800)

* Add cuda variant for mvapich2.

* Disable cuda for mvapich2 by default.

1 files changed, 13 insertions, 2 deletions

diff --git a/var/spack/repos/builtin/packages/mvapich2/package.py b/var/spack/repos/builtin/packages/mvapich2/package.py
index b20bf76e93..675731146e 100644
--- a/var/spack/repos/builtin/packages/mvapich2/package.py
+++ b/var/spack/repos/builtin/packages/mvapich2/package.py
@@ -59,6 +59,9 @@ class Mvapich2(AutotoolsPackage):
     variant('debug', default=False,
             description='Enable debug info and error messages at run-time')
 
+    variant('cuda', default=False,
+            description='Enable CUDA extension')
+
     # Accepted values are:
     #   single      - No threads (MPI_THREAD_SINGLE)
     #   funneled    - Only the main thread calls MPI (MPI_THREAD_FUNNELED)
@@ -100,9 +103,9 @@ class Mvapich2(AutotoolsPackage):
         )
     )
 
-    # FIXME : CUDA support is missing
-    depends_on('bison')
+    depends_on('bison', type='build')
     depends_on('libpciaccess', when=(sys.platform != 'darwin'))
+    depends_on('cuda', when='+cuda')
 
     def url_for_version(self, version):
         base_url = "http://mvapich.cse.ohio-state.edu/download"
@@ -217,6 +220,14 @@ class Mvapich2(AutotoolsPackage):
         else:
             args.append('--enable-fast=all')
 
+        if '+cuda' in self.spec:
+            args.extend([
+                '--enable-cuda',
+                '--with-cuda={0}'.format(spec['cuda'].prefix)
+            ])
+        else:
+            args.append('--disable-cuda')
+
         args.extend(self.process_manager_options)
         args.extend(self.network_options)
         return args