diff options
| -rw-r--r-- | var/spack/repos/builtin/packages/dealii/package.py | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/var/spack/repos/builtin/packages/dealii/package.py b/var/spack/repos/builtin/packages/dealii/package.py index d674c00098..edc73b6cfc 100644 --- a/var/spack/repos/builtin/packages/dealii/package.py +++ b/var/spack/repos/builtin/packages/dealii/package.py @@ -394,6 +394,11 @@ class Dealii(CMakePackage, CudaPackage): self.define('MPI_CXX_COMPILER', spec['mpi'].mpicxx), self.define('MPI_Fortran_COMPILER', spec['mpi'].mpifc) ]) + if '+cuda' in spec: + options.extend([ + self.define('DEAL_II_MPI_WITH_CUDA_SUPPORT', True), + self.define('CUDA_HOST_COMPILER', spec['mpi'].mpicxx) + ]) # Python bindings if spec.satisfies('@8.5.0:'): @@ -526,3 +531,8 @@ class Dealii(CMakePackage, CudaPackage): def setup_run_environment(self, env): env.set('DEAL_II_DIR', self.prefix) + + def setup_build_environment(self, env): + spec = self.spec + if '+cuda' in spec and '+mpi' in spec: + env.set('CUDAHOSTCXX', spec['mpi'].mpicxx) |