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
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
|
# 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 import *
class Arborx(CMakePackage):
"""ArborX is a performance-portable library for geometric search"""
homepage = "https://github.com/arborx/arborx"
url = "https://github.com/arborx/arborx/archive/v1.1.tar.gz"
git = "https://github.com/arborx/arborx.git"
tags = ['e4s', 'ecp']
maintainers = ['aprokop']
version('master', branch='master')
version('1.2', sha256='ed1939110b2330b7994dcbba649b100c241a2353ed2624e627a200a398096c20')
version('1.1', sha256='2b5f2d2d5cec57c52f470c2bf4f42621b40271f870b4f80cb57e52df1acd90ce')
version('1.0', sha256='9b5f45c8180622c907ef0b7cc27cb18ba272ac6558725d9e460c3f3e764f1075')
version('0.9-beta', sha256='b349b5708d1aa00e8c20c209ac75dc2d164ff9bf1b85adb5437346d194ba6c0d', deprecated=True)
# ArborX relies on Kokkos to provide devices, providing one-to-one matching
# variants. The only way to disable those devices is to make sure Kokkos
# does not provide them.
kokkos_backends = {
'serial': (True, "enable Serial backend (default)"),
'cuda': (False, "enable Cuda backend"),
'openmp': (False, "enable OpenMP backend"),
'rocm': (False, "enable HIP backend"),
'sycl': (False, "enable SYCL backend")
}
variant('mpi', default=True, description='enable MPI')
for backend in kokkos_backends:
deflt, descr = kokkos_backends[backend]
variant(backend.lower(), default=deflt, description=descr)
variant('trilinos', default=False, description='use Kokkos from Trilinos')
depends_on('cmake@3.12:', type='build')
depends_on('cmake@3.16:', type='build', when='@1.0:')
depends_on('mpi', when='+mpi')
depends_on('rocthrust', when='+rocm')
# Standalone Kokkos
depends_on('kokkos@3.1.00:', when='~trilinos')
depends_on('kokkos@3.4.00:', when='@1.2:~trilinos')
for backend in kokkos_backends:
depends_on('kokkos+%s' % backend.lower(), when='~trilinos+%s' %
backend.lower())
depends_on('kokkos+cuda_lambda', when='~trilinos+cuda')
# Trilinos/Kokkos
# Notes:
# - current version of Trilinos package does not allow disabling Serial
# - current version of Trilinos package does not allow enabling CUDA
depends_on('trilinos+kokkos', when='+trilinos')
depends_on('trilinos+openmp', when='+trilinos+openmp')
depends_on('trilinos@13.2.0:', when='@1.2:+trilinos')
conflicts('~serial', when='+trilinos')
conflicts('+cuda', when='+trilinos')
def cmake_args(self):
spec = self.spec
options = [
'-DKokkos_ROOT=%s' % (spec['kokkos'].prefix if '~trilinos' in spec
else spec['trilinos'].prefix),
self.define_from_variant('ARBORX_ENABLE_MPI', 'mpi')
]
if '+cuda' in spec:
# Only Kokkos allows '+cuda' for now
options.append(
'-DCMAKE_CXX_COMPILER=%s' % spec["kokkos"].kokkos_cxx)
if '+rocm' in spec:
options.append(
'-DCMAKE_CXX_COMPILER=%s' % spec["hip"].hipcc)
return options
examples_src_dir = "examples"
@run_after('install')
def setup_build_tests(self):
"""Copy the example source files after the package is installed to an
install test subdirectory for use during `spack test run`."""
self.cache_extra_test_sources([self.examples_src_dir])
@property
def cached_tests_work_dir(self):
"""The working directory for cached test sources."""
return join_path(self.test_suite.current_test_cache_dir,
self.examples_src_dir)
def build_tests(self):
"""Build the stand-alone/smoke test."""
arborx_dir = self.spec['arborx'].prefix
cmake_prefix_path = "-DCMAKE_PREFIX_PATH={0}".format(arborx_dir)
if '+mpi' in self.spec:
cmake_prefix_path += ";{0}".format(self.spec['mpi'].prefix)
cmake_args = [".",
cmake_prefix_path,
"-DCMAKE_CXX_COMPILER={0}".format(self.compiler.cxx),
self.define('Kokkos_ROOT', self.spec['kokkos'].prefix
if '~trilinos' in self.spec
else self.spec['trilinos'].prefix)]
self.run_test("cmake", cmake_args,
purpose="test: calling cmake",
work_dir=self.cached_tests_work_dir)
self.run_test("make", [],
purpose="test: building the tests",
work_dir=self.cached_tests_work_dir)
def test(self):
"""Perform stand-alone/smoke tests on the installed package."""
self.build_tests()
self.run_test("ctest", ["-V"],
purpose="test: running the tests",
installed=False,
work_dir=self.cached_tests_work_dir)
|
