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
|
# Copyright 2013-2018 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 Magma(CMakePackage):
"""The MAGMA project aims to develop a dense linear algebra library similar to
LAPACK but for heterogeneous/hybrid architectures, starting with current
"Multicore+GPU" systems.
"""
homepage = "http://icl.cs.utk.edu/magma/"
url = "http://icl.cs.utk.edu/projectsfiles/magma/downloads/magma-2.2.0.tar.gz"
version('2.4.0', sha256='4eb839b1295405fd29c8a6f5b4ed578476010bf976af46573f80d1169f1f9a4f')
version('2.3.0', sha256='010a4a057d7aa1e57b9426bffc0958f3d06913c9151463737e289e67dd9ea608')
version('2.2.0', sha256='df5d4ace417e5bf52694eae0d91490c6bde4cde1b0da98e8d400c5c3a70d83a2')
variant('fortran', default=True,
description='Enable Fortran bindings support')
variant('shared', default=True,
description='Enable shared library')
depends_on('blas')
depends_on('lapack')
depends_on('cuda')
conflicts('%gcc@6:', when='^cuda@:8')
conflicts('%gcc@7:', when='^cuda@:9')
patch('ibm-xl.patch', when='@2.2:%xl')
patch('ibm-xl.patch', when='@2.2:%xl_r')
patch('magma-2.3.0-gcc-4.8.patch', when='@2.3.0%gcc@:4.8')
def cmake_args(self):
spec = self.spec
options = []
options.extend([
'-DCMAKE_INSTALL_PREFIX=%s' % prefix,
'-DCMAKE_INSTALL_NAME_DIR:PATH=%s/lib' % prefix,
'-DBLAS_LIBRARIES=%s' % spec['blas'].libs.joined(';'),
# As of MAGMA v2.3.0, CMakeLists.txt does not use the variable
# BLAS_LIBRARIES, but only LAPACK_LIBRARIES, so we need to
# explicitly add blas to LAPACK_LIBRARIES.
'-DLAPACK_LIBRARIES=%s' %
(spec['lapack'].libs + spec['blas'].libs).joined(';')
])
options += ['-DBUILD_SHARED_LIBS=%s' %
('ON' if ('+shared' in spec) else 'OFF')]
if '+fortran' in spec:
options.extend([
'-DUSE_FORTRAN=yes'
])
if spec.satisfies('%xl') or spec.satisfies('%xl_r'):
options.extend([
'-DCMAKE_Fortran_COMPILER=%s' % self.compiler.f77
])
if spec.satisfies('^cuda@9.0:'):
if '@:2.2.0' in spec:
options.extend(['-DGPU_TARGET=sm30'])
else:
options.extend(['-DGPU_TARGET=sm_30'])
return options
@run_after('install')
def post_install(self):
install('magmablas/atomics.cuh', self.prefix.include)
install('control/magma_threadsetting.h', self.prefix.include)
install('control/pthread_barrier.h', self.prefix.include)
install('control/magma_internal.h', self.prefix.include)
|
