hypre icon indicating copy to clipboard operation
hypre copied to clipboard

Hypre spack support for umpire, gpu-aware-mpi

Open PaulMullowney opened this issue 2 years ago • 0 comments

I've been able to get gpu-aware-mpi working through spack builds. Umpire seems a little trickier. Any help on this would be appreciated.

diff --git a/var/spack/repos/builtin/packages/hypre/package.py b/var/spack/repos/builtin/packages/hypre/package.py
index b38ae55ab0..339147acfb 100644
--- a/var/spack/repos/builtin/packages/hypre/package.py
+++ b/var/spack/repos/builtin/packages/hypre/package.py
@@ -67,6 +67,7 @@ class Hypre(AutotoolsPackage, CudaPackage, ROCmPackage):
     variant('debug', default=False,
             description='Build debug instead of optimized version')
     variant('unified-memory', default=False, description='Use unified memory')
+    variant('gpu-aware-mpi', default=False, description='Use gpu-aware mpi')
     variant('fortran', default=True,
             description='Enables fortran bindings')
     variant('gptune', default=False,
@@ -226,6 +227,9 @@ def configure_args(self):
         if '+unified-memory' in spec:
             configure_args.append('--enable-unified-memory')

+        if '+gpu-aware-mpi' in spec:
+            configure_args.append('--enable-gpu-aware-mpi')
+

-Paul

PaulMullowney avatar Jul 15 '22 21:07 PaulMullowney