1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
|
# 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')
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("+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')
# 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
|