diff options
Diffstat (limited to 'var/spack/repos/builtin/packages/cabana/package.py')
-rw-r--r-- | var/spack/repos/builtin/packages/cabana/package.py | 65 |
1 files changed, 47 insertions, 18 deletions
diff --git a/var/spack/repos/builtin/packages/cabana/package.py b/var/spack/repos/builtin/packages/cabana/package.py index ff55a5661a..9d2b83ff5d 100644 --- a/var/spack/repos/builtin/packages/cabana/package.py +++ b/var/spack/repos/builtin/packages/cabana/package.py @@ -4,6 +4,7 @@ # SPDX-License-Identifier: (Apache-2.0 OR MIT) from spack import * +from spack.pkg.builtin.kokkos import Kokkos class Cabana(CMakePackage): @@ -11,42 +12,70 @@ class Cabana(CMakePackage): """ homepage = "https://github.com/ECP-copa/Cabana" git = "https://github.com/ECP-copa/Cabana.git" - url = "https://github.com/ECP-copa/Cabana/archive/0.1.0.tar.gz" + url = "https://github.com/ECP-copa/Cabana/archive/0.4.0.tar.gz" + + maintainers = ["junghans", "sslattery", "streeve"] version('master', branch='master') + version('0.4.0', sha256='c347d23dc4a5204f9cc5906ccf3454f0b0b1612351bbe0d1c58b14cddde81e85') version('0.3.0', sha256='fb67ab9aaf254b103ae0eb5cc913ddae3bf3cd0cf6010e9686e577a2981ca84f') version('0.2.0', sha256='3e0c0e224e90f4997f6c7e2b92f00ffa18f8bcff72f789e0908cea0828afc2cb') version('0.1.0', sha256='3280712facf6932b9d1aff375b24c932abb9f60a8addb0c0a1950afd0cb9b9cf') version('0.1.0-rc0', sha256='73754d38aaa0c2a1e012be6959787108fec142294774c23f70292f59c1bdc6c5') - variant('serial', default=True, description="enable Serial backend (default)") - variant('openmp', default=False, description="enable OpenMP backend") - variant('cuda', default=False, description="enable Cuda backend") + _kokkos_backends = Kokkos.devices_variants + for _backend in _kokkos_backends: + _deflt, _descr = _kokkos_backends[_backend] + variant(_backend.lower(), default=_deflt, description=_descr) + variant('shared', default=True, description='Build shared libraries') variant('mpi', default=True, description='Build with mpi support') + variant('arborx', default=False, description='Build with ArborX support') + variant('heffte', default=False, description='Build with heFFTe support') + variant('hypre', default=False, description='Build with HYPRE support') depends_on("cmake@3.9:", type='build') - depends_on("kokkos-legacy+serial", when="@:0.2.0+serial") - depends_on("kokkos-legacy+openmp", when="@:0.2.0+openmp") - depends_on("kokkos-legacy+cuda", when="@:0.2.0+cuda") - depends_on("kokkos@3.1:+serial", when="@0.3.0:+serial") - depends_on("kokkos@3.1:+openmp", when="@0.3.0:+openmp") - depends_on("kokkos@3.1:+cuda", when="@0.3.0:+cuda") + _versions = { + ":0.2.0": "-legacy", + "0.3.0": "@3.1:", + "0.4.0": "@3.2:" + } + for _version in _versions: + _kk_version = _versions[_version] + for _backend in _kokkos_backends: + if (_kk_version == "-legacy" and _backend == 'pthread'): + _kk_spec = 'kokkos-legacy+pthreads' + elif (_kk_version == "-legacy" and + _backend not in ['serial', 'openmp', 'cuda']): + continue + else: + _kk_spec = 'kokkos{0}+{1}'.format(_kk_version, _backend) + depends_on(_kk_spec, when='@{0}+{1}'.format(_version, _backend)) + depends_on("arborx", when="@0.3.0:+arborx") + depends_on("heffte@2.0:", when="@0.4.0:+heffte") + depends_on("hypre-cmake@2.22.0:", when="@0.4.0:+hypre") depends_on('mpi', when='+mpi') + conflicts("+rocm", when="@:0.2.0") + conflicts("+sycl", when="@:0.3.0") + def cmake_args(self): options = [ '-DCabana_ENABLE_TESTING=ON', - '-DCabana_ENABLE_Serial=%s' % ( - 'On' if '+serial' in self.spec else 'Off'), - '-DCabana_ENABLE_OpenMP=%s' % ( - 'On' if '+openmp' in self.spec else 'Off'), - '-DCabana_ENABLE_Cuda=%s' % ( - 'On' if '+cuda' in self.spec else 'Off'), - '-DCabana_ENABLE_MPI=%s' % ( - 'On' if '+mpi' in self.spec else 'Off'), '-DBUILD_SHARED_LIBS=%s' % ( 'On' if '+shared' in self.spec else 'Off') ] + # These variables were removed in 0.3.0 (where backends are + # automatically used from Kokkos) + if self.spec.satisfies('@:0.2.0'): + backends = {'serial': 'Serial', + 'openmp': 'OpenMP', + 'cuda': 'Cuda'} + for backend in backends: + cbn_option = 'Cabana_ENABLE_' + backends[backend] + options.append(self.define_from_variant(cbn_option, backend)) + + if self.spec.satisfies('@:0.3.0'): + options.append(self.define_from_variant('Cabana_ENABLE_MPI', 'mpi')) return options |