hypre
hypre copied to clipboard
Hypre spack support for umpire, gpu-aware-mpi
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