zhangyiss/spack
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

amrex: support sundials variant in newer amrex versions (#25745)

Browse Source 
* amrex: support sundials variant in newer amrex versions

* propagate cuda_arch to sundials

* change to old string formatting

* require sundials+rocm when amrex+rocm
This commit is contained in:
Cody Balos 2021-09-07 18:05:32 -07:00 committed by GitHub
parent c6e538583f
commit 32b6da8d57
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
var/spack/repos/builtin/packages/amrex/package.py
View File

@ -69,20 +69,25 @@ class Amrex(CMakePackage, CudaPackage, ROCmPackage):
description='Build particle classes') description='Build particle classes')
variant('plotfile_tools', default=False, variant('plotfile_tools', default=False,
description='Build plotfile_tools like fcompare') description='Build plotfile_tools like fcompare')
variant('sundials', default=False,
description='Build AMReX with SUNDIALS support')
variant('hdf5', default=False, variant('hdf5', default=False,
description='Enable HDF5-based I/O') description='Enable HDF5-based I/O')
variant('hypre', default=False, variant('hypre', default=False,
description='Enable Hypre interfaces') description='Enable Hypre interfaces')
variant('petsc', default=False, variant('petsc', default=False,
description='Enable PETSc interfaces') description='Enable PETSc interfaces')
variant('sundials', default=False,
description='Enable SUNDIALS interfaces')
variant('pic', default=False, variant('pic', default=False,
description='Enable PIC') description='Enable PIC')
# Build dependencies # Build dependencies
depends_on('mpi', when='+mpi') depends_on('mpi', when='+mpi')
depends_on('sundials@4.0.0:4.1.0 +ARKODE +CVODE', when='@19.08: +sundials') depends_on('sundials@4.0.0:4.1.0 +ARKODE +CVODE', when='@19.08:20.11 +sundials')
depends_on('sundials@5.7.0: +ARKODE +CVODE', when='@21.07: +sundials')
for arch in CudaPackage.cuda_arch_values:
depends_on('sundials@5.7.0: +ARKODE +CVODE +cuda cuda_arch=%s' % arch, when='@21.07: +sundials +cuda cuda_arch=%s' % arch)
for tgt in ROCmPackage.amdgpu_targets:
depends_on('sundials@5.7.0: +ARKODE +CVODE +rocm amdgpu_target=%s' % tgt, when='@21.07: +sundials +rocm amdgpu_target=%s' % tgt)
depends_on('cuda@9.0.0:', when='+cuda') depends_on('cuda@9.0.0:', when='+cuda')
depends_on('python@2.7:', type='build', when='@:20.04') depends_on('python@2.7:', type='build', when='@:20.04')
depends_on('cmake@3.5:', type='build', when='@:18.10.99') depends_on('cmake@3.5:', type='build', when='@:18.10.99')
@ -104,10 +109,10 @@ class Amrex(CMakePackage, CudaPackage, ROCmPackage):
conflicts('%gcc@8.1.0:8.2.0', when='@21.01:21.02') conflicts('%gcc@8.1.0:8.2.0', when='@21.01:21.02')
# Check options compatibility # Check options compatibility
conflicts('+sundials', when='~fortran', conflicts('+sundials', when='@19.08:20.11 ~fortran',
msg='AMReX SUNDIALS support needs AMReX Fortran API (+fortran)') msg='AMReX SUNDIALS support needs AMReX Fortran API (+fortran)')
conflicts('+sundials', when='@20.12:', conflicts('+sundials', when='@20.12:21.06',
msg='AMReX >= 20.12 no longer supports SUNDIALS interfaces') msg='AMReX 20.12 -- 21.06 does not support SUNDIALS interfaces')
conflicts('+hdf5', when='@:20.06', conflicts('+hdf5', when='@:20.06',
msg='AMReX HDF5 support needs AMReX newer than version 20.06') msg='AMReX HDF5 support needs AMReX newer than version 20.06')
conflicts('+hypre', when='@:20.06', conflicts('+hypre', when='@:20.06',
@ -176,6 +181,7 @@ def cmake_args(self):
self.define_from_variant('AMReX_HDF5', 'hdf5'), self.define_from_variant('AMReX_HDF5', 'hdf5'),
self.define_from_variant('AMReX_HYPRE', 'hypre'), self.define_from_variant('AMReX_HYPRE', 'hypre'),
self.define_from_variant('AMReX_PETSC', 'petsc'), self.define_from_variant('AMReX_PETSC', 'petsc'),
self.define_from_variant('AMReX_SUNDIALS', 'sundials'),
self.define_from_variant('AMReX_PIC', 'pic'), self.define_from_variant('AMReX_PIC', 'pic'),
] ]