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
|
# Copyright 2013-2020 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 Rocalution(CMakePackage):
"""rocALUTION is a sparse linear algebra library with focus on
exploring fine-grained parallelism on top of AMD's Radeon Open
eCosystem Platform ROCm runtime and toolchains, targeting modern
CPU and GPU platforms. Based on C++ and HIP, it provides a portable,
generic and flexible design that allows seamless integration with
other scientific software packages."""
homepage = "https://github.com/ROCmSoftwarePlatform/rocALUTION"
url = "https://github.com/ROCmSoftwarePlatform/rocALUTION/archive/rocm-3.7.0.tar.gz"
maintainers = ['srekolam', 'arjun-raj-kuppala']
version('3.10.0', sha256='c24cb9d1a8a1a3118040b8b16dec7c06268bcf157424d3378256cc9eb93f1b58')
version('3.9.0', sha256='1ce36801fe1d44f743b46b43345c0cd90d76b73911b2ec97be763f93a35396fb')
version('3.8.0', sha256='39e64a29e75c4276163a93596436064c6338770ca72ce7f43711ed8285ed2de5')
version('3.7.0', sha256='4d6b20aaaac3bafb7ec084d684417bf578349203b0f9f54168f669e3ec5699f8')
version('3.5.0', sha256='be2f78c10c100d7fd9df5dd2403a44700219c2cbabaacf2ea50a6e2241df7bfe')
depends_on('cmake@3.5:', type='build')
for ver in ['3.5.0', '3.7.0', '3.8.0', '3.9.0', '3.10.0']:
depends_on('hip@' + ver, when='@' + ver)
depends_on('rocblas@' + ver, type='link', when='@' + ver)
depends_on('rocprim@' + ver, type='link', when='@' + ver)
depends_on('rocsparse@' + ver, type='link', when='@' + ver)
depends_on('rocm-device-libs@' + ver, type='build', when='@' + ver)
depends_on('comgr@' + ver, type='build', when='@' + ver)
depends_on('llvm-amdgpu@' + ver, type='build', when='@' + ver)
if ver in ['3.9.0', '3.10.0']:
depends_on('rocrand@' + ver, type='link', when='@' + ver)
patch('0001-fix-hip-build-error.patch')
def setup_build_environment(self, env):
env.set('CXX', self.spec['hip'].hipcc)
def patch(self):
if '@3.9.0:' in self.spec:
kwargs = {'ignore_absent': False, 'backup': False, 'string': False}
with working_dir('src/base/hip'):
match = '^#include <rocrand/rocrand.hpp>'
substitute = "#include <rocrand.hpp>"
files = ['hip_rand_normal.hpp', 'hip_rand_uniform.hpp']
filter_file(match, substitute, *files, **kwargs)
def cmake_args(self):
args = [
'-DSUPPORT_HIP=ON',
'-DSUPPORT_MPI=OFF',
'-DBUILD_CLIENTS_SAMPLES=OFF'
]
return args
|
