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
|
# 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 Slate(CMakePackage, CudaPackage, ROCmPackage):
"""The Software for Linear Algebra Targeting Exascale (SLATE) project is
to provide fundamental dense linear algebra capabilities to the US
Department of Energy and to the high-performance computing (HPC) community
at large. To this end, SLATE will provide basic dense matrix operations
(e.g., matrix multiplication, rank-k update, triangular solve), linear
systems solvers, least square solvers, singular value and eigenvalue
solvers."""
homepage = "https://icl.utk.edu/slate/"
git = "https://bitbucket.org/icl/slate"
url = 'https://bitbucket.org/icl/slate/downloads/slate-2020.10.00.tar.gz'
maintainers = ['G-Ragghianti', 'mgates3']
test_requires_compiler = True
version('master', branch='master')
version('2021.05.02', sha256='29667a9e869e41fbc22af1ae2bcd425d79b4094bbb3f21c411888e7adc5d12e3')
version('2021.05.01', sha256='d9db2595f305eb5b1b49a77cc8e8c8e43c3faab94ed910d8387c221183654218')
version('2020.10.00', sha256='ff58840cdbae2991d100dfbaf3ef2f133fc2f43fc05f207dc5e38a41137882ab')
variant('mpi', default=True, description='Build with MPI support (without MPI is experimental).')
variant('openmp', default=True, description='Build with OpenMP support.')
variant('shared', default=True, description='Build shared library')
depends_on('mpi', when='+mpi')
depends_on('blas')
depends_on('blaspp ~cuda', when='~cuda')
depends_on('blaspp +cuda', when='+cuda')
depends_on('blaspp ~rocm', when='~rocm')
for val in ROCmPackage.amdgpu_targets:
depends_on('blaspp +rocm amdgpu_target=%s' % val, when='amdgpu_target=%s' % val)
depends_on('lapackpp@2021.04.00:', when='@2021.05.01:')
depends_on('lapackpp@2020.10.02', when='@2020.10.00')
depends_on('lapackpp@master', when='@master')
depends_on('scalapack')
cpp_17_msg = 'Requires C++17 compiler support'
conflicts('%gcc@:5', msg=cpp_17_msg)
conflicts('%xl', msg=cpp_17_msg)
conflicts('%xl_r', msg=cpp_17_msg)
conflicts('%intel@19:', msg='Does not currently build with icpc >= 2019')
conflicts('+rocm', when='@:2020.10.00', msg='ROCm support requires SLATE 2021.05.01 or greater')
conflicts('+rocm', when='+cuda', msg='SLATE only supports one GPU backend at a time')
def cmake_args(self):
spec = self.spec
backend_config = '-Duse_cuda=%s' % ('+cuda' in spec)
if self.version >= Version('2021.05.01'):
backend = 'none'
if '+cuda' in spec:
backend = 'cuda'
if '+rocm' in spec:
backend = 'hip'
backend_config = '-Dgpu_backend=%s' % backend
return [
'-Dbuild_tests=%s' % self.run_tests,
'-Duse_openmp=%s' % ('+openmp' in spec),
'-DBUILD_SHARED_LIBS=%s' % ('+shared' in spec),
backend_config,
'-Duse_mpi=%s' % ('+mpi' in spec),
'-DSCALAPACK_LIBRARIES=%s' % spec['scalapack'].libs.joined(';')
]
examples_src_dir = 'examples'
@run_after('install')
def cache_test_sources(self):
"""Copy the example source files after the package is installed to an
install test subdirectory for use during `spack test run`."""
if self.spec.satisfies('@master'):
self.cache_extra_test_sources([self.examples_src_dir])
def test(self):
if not self.spec.satisfies('@master') or '+mpi' not in self.spec:
print('Skipping: stand-alone tests only run on master with +mpi')
return
test_dir = join_path(self.install_test_root, self.examples_src_dir)
test_bld_dir = join_path(test_dir, 'build')
with working_dir(test_bld_dir, create=True):
cmake('..')
make()
test_args = ['-n', '4', './ex05_blas']
mpiexe_f = which('srun', 'mpirun', 'mpiexec')
if mpiexe_f:
self.run_test(mpiexe_f.command, test_args,
purpose='SLATE smoke test')
make('clean')
|