201 lines
8.1 KiB
Python
201 lines
8.1 KiB
Python
# 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)
|
|
|
|
import socket
|
|
|
|
from spack import *
|
|
|
|
|
|
class Raja(CachedCMakePackage, CudaPackage, ROCmPackage):
|
|
"""RAJA Parallel Framework."""
|
|
|
|
homepage = "https://software.llnl.gov/RAJA/"
|
|
git = "https://github.com/LLNL/RAJA.git"
|
|
tags = ['radiuss', 'e4s']
|
|
|
|
maintainers = ['davidbeckingsale']
|
|
|
|
version('develop', branch='develop', submodules='True')
|
|
version('main', branch='main', submodules='True')
|
|
version('0.14.0', tag='v0.14.0', submodules='True')
|
|
version('0.13.0', tag='v0.13.0', submodules='True')
|
|
version('0.12.1', tag='v0.12.1', submodules="True")
|
|
version('0.12.0', tag='v0.12.0', submodules="True")
|
|
version('0.11.0', tag='v0.11.0', submodules="True")
|
|
version('0.10.1', tag='v0.10.1', submodules="True")
|
|
version('0.10.0', tag='v0.10.0', submodules="True")
|
|
version('0.9.0', tag='v0.9.0', submodules="True")
|
|
version('0.8.0', tag='v0.8.0', submodules="True")
|
|
version('0.7.0', tag='v0.7.0', submodules="True")
|
|
version('0.6.0', tag='v0.6.0', submodules="True")
|
|
version('0.5.3', tag='v0.5.3', submodules="True")
|
|
version('0.5.2', tag='v0.5.2', submodules="True")
|
|
version('0.5.1', tag='v0.5.1', submodules="True")
|
|
version('0.5.0', tag='v0.5.0', submodules="True")
|
|
version('0.4.1', tag='v0.4.1', submodules="True")
|
|
version('0.4.0', tag='v0.4.0', submodules="True")
|
|
|
|
# export targets when building pre-2.4.0 release with BLT 0.4.0+
|
|
patch('https://github.com/LLNL/RAJA/commit/eca1124ee4af380d6613adc6012c307d1fd4176b.patch',
|
|
sha256='57dd531a50ac791b4bb214d34a4bf3fca1349354927c72915b7ccd20524701a9',
|
|
when='@:0.13.0 ^blt@0.4:')
|
|
|
|
variant('openmp', default=True, description='Build OpenMP backend')
|
|
variant('shared', default=True, description='Build Shared Libs')
|
|
variant('examples', default=True, description='Build examples.')
|
|
variant('exercises', default=True, description='Build exercises.')
|
|
# TODO: figure out gtest dependency and then set this default True
|
|
# and remove the +tests conflict below.
|
|
variant('tests', default=False, description='Build tests')
|
|
|
|
depends_on('blt')
|
|
depends_on('blt@0.4.1:', type='build', when='@0.14.0:')
|
|
depends_on('blt@0.4.0:', type='build', when='@0.13.0')
|
|
depends_on('blt@0.3.6:', type='build', when='@:0.12.0')
|
|
|
|
depends_on('camp@0.2.2', when='@0.14.0:')
|
|
depends_on('camp@0.1.0', when='@0.12.0:0.13.0')
|
|
|
|
with when('+rocm @0.12.0:'):
|
|
depends_on('camp+rocm')
|
|
for arch in ROCmPackage.amdgpu_targets:
|
|
depends_on('camp+rocm amdgpu_target={0}'.format(arch),
|
|
when='amdgpu_target={0}'.format(arch))
|
|
conflicts('+openmp')
|
|
|
|
with when('+cuda @0.12.0:'):
|
|
depends_on('camp+cuda')
|
|
for sm_ in CudaPackage.cuda_arch_values:
|
|
depends_on('camp +cuda cuda_arch={0}'.format(sm_),
|
|
when='cuda_arch={0}'.format(sm_))
|
|
|
|
def _get_sys_type(self, spec):
|
|
sys_type = spec.architecture
|
|
if "SYS_TYPE" in env:
|
|
sys_type = env["SYS_TYPE"]
|
|
return sys_type
|
|
|
|
@property
|
|
def cache_name(self):
|
|
hostname = socket.gethostname()
|
|
if "SYS_TYPE" in env:
|
|
hostname = hostname.rstrip('1234567890')
|
|
return "{0}-{1}-{2}@{3}.cmake".format(
|
|
hostname,
|
|
self._get_sys_type(self.spec),
|
|
self.spec.compiler.name,
|
|
self.spec.compiler.version
|
|
)
|
|
|
|
def initconfig_hardware_entries(self):
|
|
spec = self.spec
|
|
entries = super(Raja, self).initconfig_hardware_entries()
|
|
|
|
entries.append(cmake_cache_option("ENABLE_OPENMP", '+openmp' in spec))
|
|
|
|
if '+cuda' in spec:
|
|
entries.append(cmake_cache_option("ENABLE_CUDA", True))
|
|
|
|
if not spec.satisfies('cuda_arch=none'):
|
|
cuda_arch = spec.variants['cuda_arch'].value
|
|
entries.append(cmake_cache_string(
|
|
"CUDA_ARCH", 'sm_{0}'.format(cuda_arch[0])))
|
|
entries.append(cmake_cache_string(
|
|
"CMAKE_CUDA_ARCHITECTURES", '{0}'.format(cuda_arch[0])))
|
|
else:
|
|
entries.append(cmake_cache_option("ENABLE_CUDA", False))
|
|
|
|
if '+rocm' in spec:
|
|
entries.append(cmake_cache_option("ENABLE_HIP", True))
|
|
entries.append(cmake_cache_path(
|
|
"HIP_ROOT_DIR", '{0}'.format(spec['hip'].prefix)))
|
|
archs = self.spec.variants['amdgpu_target'].value
|
|
if archs != 'none':
|
|
arch_str = ",".join(archs)
|
|
entries.append(cmake_cache_string(
|
|
"HIP_HIPCC_FLAGS", '--amdgpu-target={0}'.format(arch_str)))
|
|
else:
|
|
entries.append(cmake_cache_option("ENABLE_HIP", False))
|
|
|
|
return entries
|
|
|
|
def initconfig_package_entries(self):
|
|
spec = self.spec
|
|
entries = []
|
|
|
|
entries.append(cmake_cache_path("BLT_SOURCE_DIR", spec['blt'].prefix))
|
|
entries.append(cmake_cache_path("camp_DIR", spec['camp'].prefix))
|
|
entries.append(cmake_cache_option("BUILD_SHARED_LIBS", '+shared' in spec))
|
|
entries.append(cmake_cache_option("ENABLE_EXAMPLES", '+examples' in spec))
|
|
if spec.satisfies('@0.14.0:'):
|
|
entries.append(cmake_cache_option("RAJA_ENABLE_EXERCISES",
|
|
'+exercises' in spec))
|
|
else:
|
|
entries.append(cmake_cache_option("ENABLE_EXERCISES",
|
|
'+exercises' in spec))
|
|
|
|
# Work around spack adding -march=ppc64le to SPACK_TARGET_ARGS which
|
|
# is used by the spack compiler wrapper. This can go away when BLT
|
|
# removes -Werror from GTest flags
|
|
if self.spec.satisfies('%clang target=ppc64le:') or not self.run_tests:
|
|
entries.append(cmake_cache_option("ENABLE_TESTS", False))
|
|
else:
|
|
entries.append(cmake_cache_option("ENABLE_TESTS", True))
|
|
|
|
return entries
|
|
|
|
def cmake_args(self):
|
|
options = []
|
|
return options
|
|
|
|
@property
|
|
def build_relpath(self):
|
|
"""Relative path to the cmake build subdirectory."""
|
|
return join_path('..', self.build_dirname)
|
|
|
|
@run_after('install')
|
|
def setup_build_tests(self):
|
|
"""Copy the build test files after the package is installed to a
|
|
relative install test subdirectory for use during `spack test run`."""
|
|
# Now copy the relative files
|
|
self.cache_extra_test_sources(self.build_relpath)
|
|
|
|
# Ensure the path exists since relying on a relative path at the
|
|
# same level as the normal stage source path.
|
|
mkdirp(self.install_test_root)
|
|
|
|
@property
|
|
def _extra_tests_path(self):
|
|
# TODO: The tests should be converted to re-build and run examples
|
|
# TODO: using the installed libraries.
|
|
return join_path(self.install_test_root, self.build_relpath, 'bin')
|
|
|
|
def _test_examples(self):
|
|
"""Perform very basic checks on a subset of copied examples."""
|
|
checks = [
|
|
('ex5_line-of-sight_solution',
|
|
[r'RAJA sequential', r'RAJA OpenMP', r'result -- PASS']),
|
|
('ex6_stencil-offset-layout_solution',
|
|
[r'RAJA Views \(permuted\)', r'result -- PASS']),
|
|
('ex8_tiled-matrix-transpose_solution',
|
|
[r'parallel top inner loop',
|
|
r'collapsed inner loops', r'result -- PASS']),
|
|
('kernel-dynamic-tile', [r'Running index', r'(24,24)']),
|
|
('plugin-example',
|
|
[r'Launching host kernel for the 10 time']),
|
|
('tut_batched-matrix-multiply', [r'result -- PASS']),
|
|
('wave-eqn', [r'Max Error = 2', r'Evolved solution to time'])
|
|
]
|
|
for exe, expected in checks:
|
|
reason = 'test: checking output of {0} for {1}' \
|
|
.format(exe, expected)
|
|
self.run_test(exe, [], expected, installed=False,
|
|
purpose=reason, skip_missing=True,
|
|
work_dir=self._extra_tests_path)
|
|
|
|
def test(self):
|
|
"""Perform smoke tests."""
|
|
self._test_examples()
|