# Copyright 2013-2021 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/v0.9-beta.tar.gz"
git = "https://github.com/arborx/arborx.git"
maintainers = ['aprokop']
version('master', branch='master')
version('0.9-beta', sha256='b349b5708d1aa00e8c20c209ac75dc2d164ff9bf1b85adb5437346d194ba6c0d')
# 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"),
'hip': (False, "enable HIP 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('mpi', when='+mpi')
# Standalone Kokkos
depends_on('kokkos@3.1.00:', when='~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:
# - there is no Trilinos release with Kokkos 3.1 yet
# - current version of Trilinos package does not allow disabling Serial
# - current version of Trilinos package does not allow enabling CUDA
depends_on('trilinos+kokkos@develop', when='+trilinos')
depends_on('trilinos+openmp', when='+trilinos+openmp')
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),
'-DARBORX_ENABLE_MPI=%s' % ('ON' if '+mpi' in spec else 'OFF')
]
if '+cuda' in spec:
# Only Kokkos allows '+cuda' for now
options.append(
'-DCMAKE_CXX_COMPILER=%s' % spec["kokkos"].kokkos_cxx)
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])
def build_tests(self):
"""Build test."""
cmake_build_path = join_path(self.install_test_root,
self.examples_src_dir, "build")
mkdirp(cmake_build_path)
cmake_prefix_path = "-DCMAKE_PREFIX_PATH={0}".format(self.spec['arborx'].prefix)
# We don't need to append the path to Kokkos to CMAKE_PREFIX_PATH
# since a hint is already hardcoded inside the CMake ArborX configuration.
# Omitting it here allows us to avoid to distinguish between Kokkos
# being installed as a standalone or as part of Trilinos.
if '+mpi' in self.spec:
cmake_prefix_path += ";{0}".format(self.spec['mpi'].prefix)
with working_dir(cmake_build_path):
cmake_args = ["..",
cmake_prefix_path,
"-DCMAKE_CXX_COMPILER={0}".format(self.compiler.cxx)]
cmake(*cmake_args)
make()
def run_tests(self):
"""Run test."""
reason = 'Checking ability to execute.'
run_path = join_path(self.install_test_root, self.examples_src_dir, 'build')
with working_dir(run_path):
self.run_test('ctest', ['-V'], [], installed=False, purpose=reason)
def test(self):
self.build_tests()
self.run_tests()