5b4c076ddb
* cabana: add cajita variant * Update var/spack/repos/builtin/packages/cabana/package.py Co-authored-by: Sam Reeve <6740307+streeve@users.noreply.github.com> * cabana: cannot build +cajita ~mpi Co-authored-by: Sam Reeve <6740307+streeve@users.noreply.github.com>
103 lines
4.3 KiB
Python
103 lines
4.3 KiB
Python
# Copyright 2013-2022 Lawrence Livermore National Security, LLC and other
|
|
# Spack Project Developers. See the top-level COPYRIGHT file for details.
|
|
#
|
|
# SPDX-License-Identifier: (Apache-2.0 OR MIT)
|
|
|
|
from spack.package import *
|
|
from spack.pkg.builtin.kokkos import Kokkos
|
|
|
|
|
|
class Cabana(CMakePackage):
|
|
"""The Exascale Co-Design Center for Particle Applications Toolkit
|
|
"""
|
|
homepage = "https://github.com/ECP-copa/Cabana"
|
|
git = "https://github.com/ECP-copa/Cabana.git"
|
|
url = "https://github.com/ECP-copa/Cabana/archive/0.4.0.tar.gz"
|
|
|
|
maintainers = ["junghans", "streeve", "sslattery"]
|
|
|
|
tags = ['e4s', 'ecp']
|
|
|
|
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')
|
|
|
|
_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')
|
|
variant('cajita', default=False, description='Build Cajita subpackage')
|
|
|
|
depends_on("cmake@3.9:", type='build')
|
|
depends_on("googletest", type='build')
|
|
_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("hypre-cmake@2.22.0:", when="@0.4.0:+hypre")
|
|
depends_on("hypre-cmake@2.22.1:", when="@master+hypre")
|
|
# Heffte pinned at 2.0.0 for now because its cmakefiles can't roll forward
|
|
# compatibilty from 2.0.0 to later minor versions and cabana cmakefiles are
|
|
# currently requesting version 2.0.0. Will be fixed for later cabana versions
|
|
# as cabana requested version number and heffte version compatibility changes
|
|
depends_on("heffte@2.0.0", when="@0.4.0+heffte")
|
|
depends_on("heffte@2.1.0", when="@master+heffte")
|
|
depends_on('mpi', when='+mpi')
|
|
|
|
conflicts("+cajita ~mpi")
|
|
|
|
conflicts("+rocm", when="@:0.2.0")
|
|
conflicts("+sycl", when="@:0.3.0")
|
|
|
|
def cmake_args(self):
|
|
options = [
|
|
'-DCabana_ENABLE_TESTING=ON',
|
|
'-DBUILD_SHARED_LIBS=%s' % (
|
|
'On' if '+shared' in self.spec else 'Off')
|
|
]
|
|
|
|
# Enable Cabana submodules based on flags above
|
|
if '+hypre' in self.spec:
|
|
options.append('-DCabana_REQUIRE_HYPRE=ON')
|
|
if '+heffte' in self.spec:
|
|
options.append('-DCabana_REQUIRE_HEFFTE=ON')
|
|
|
|
options.append(self.define_from_variant('Cabana_ENABLE_CAJITA', 'cajita'))
|
|
|
|
# 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
|