zhangyiss/spack
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add strumpack v5.1.0, add support for ROCm/HIP (#21203)

Browse Source
This commit is contained in:
Pieter Ghysels 2021-01-23 00:31:23 -08:00 committed by GitHub
parent 12eb4a146f
commit dba6a26920
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 25 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
var/spack/repos/builtin/packages/strumpack/package.py
View File

@ -6,7 +6,7 @@
from spack import *
class Strumpack(CMakePackage, CudaPackage):
class Strumpack(CMakePackage, CudaPackage, ROCmPackage):
"""STRUMPACK -- STRUctured Matrix PACKage - provides linear solvers
for sparse matrices and for dense rank-structured matrices, i.e.,
matrices that exhibit some kind of low-rank property. It provides a
@ -18,12 +18,13 @@ class Strumpack(CMakePackage, CudaPackage):
iterative solvers."""
homepage = "http://portal.nersc.gov/project/sparse/strumpack"
url = "https://github.com/pghysels/STRUMPACK/archive/v4.0.0.tar.gz"
url = "https://github.com/pghysels/STRUMPACK/archive/v5.1.0.tar.gz"
git = "https://github.com/pghysels/STRUMPACK.git"
maintainers = ['pghysels']
version('master', branch='master')
version('5.1.1', sha256='6cf4eaae5beb9bd377f2abce9e4da9fd3e95bf086ae2f04554fad6dd561c28b9')
version('5.0.0', sha256='bdfd1620ff7158d96055059be04ee49466ebaca8213a2fdab33e2d4571019a49')
version('4.0.0', sha256='a3629f1f139865c74916f8f69318f53af6319e7f8ec54e85c16466fd7d256938')
version('3.3.0', sha256='499fd3b58656b4b6495496920e5372895861ebf15328be8a7a9354e06c734bc7')
@ -34,8 +35,6 @@ class Strumpack(CMakePackage, CudaPackage):
variant('mpi', default=True, description='Use MPI')
variant('openmp', default=True,
description='Enable thread parallellism via tasking with OpenMP')
variant('cuda', default=True,
description='Enable CUDA support')
variant('parmetis', default=True,
description='Enable use of ParMetis')
variant('scotch', default=False,
@ -70,12 +69,16 @@ class Strumpack(CMakePackage, CudaPackage):
depends_on('butterflypack@1.2.0:', when='@4.0.0: +butterflypack+mpi')
depends_on('cuda', when='@4.0.0: +cuda')
depends_on('zfp', when='+zfp')
depends_on('hipblas', when='+rocm')
depends_on('rocsolver', when='+rocm')
conflicts('+parmetis', when='~mpi')
conflicts('+butterflypack', when='~mpi')
conflicts('+butterflypack', when='@:3.2.0')
conflicts('+cuda', when='@:3.9.999')
conflicts('+zfp', when='@:3.9.999')
conflicts('+cuda', when='@:3.9.999')
conflicts('+rocm', when='@:5.0.999')
conflicts('+rocm', when='+cuda')
patch('intel-19-compile.patch', when='@3.1.1')
@ -88,6 +91,8 @@ def on_off(varstr):
args = [
'-DSTRUMPACK_USE_MPI=%s' % on_off('+mpi'),
'-DSTRUMPACK_USE_OPENMP=%s' % on_off('+openmp'),
'-DSTRUMPACK_USE_CUDA=%s' % on_off('+cuda'),
'-DSTRUMPACK_USE_HIP=%s' % on_off('+rocm'),
'-DTPL_ENABLE_PARMETIS=%s' % on_off('+parmetis'),
'-DTPL_ENABLE_SCOTCH=%s' % on_off('+scotch'),
'-DTPL_ENABLE_BPACK=%s' % on_off('+butterflypack'),
@ -99,6 +104,7 @@ def on_off(varstr):
'-DTPL_LAPACK_LIBRARIES=%s' % spec['lapack'].libs.joined(";"),
'-DTPL_SCALAPACK_LIBRARIES=%s' % spec['scalapack'].
libs.joined(";"),
'-DBUILD_SHARED_LIBS=%s' % on_off('+shared')
]
if spec.satisfies('@:3.9.999'):
@ -112,13 +118,21 @@ def on_off(varstr):
'-DSTRUMPACK_C_INTERFACE=%s' % on_off('+c_interface'),
])
if spec.satisfies('@4.0.0:'):
if '+cuda' in spec:
args.extend([
'-DSTRUMPACK_USE_CUDA=%s' % on_off('+cuda')
])
'-DCUDA_TOOLKIT_ROOT_DIR={0}'.format(spec['cuda'].prefix),
'-DCMAKE_CUDA_HOST_COMPILER={0}'.format(env["SPACK_CXX"])])
cuda_archs = spec.variants['cuda_arch'].value
if 'none' not in cuda_archs:
args.append('-DCUDA_NVCC_FLAGS={0}'.
format(' '.join(self.cuda_flags(cuda_archs))))
args.extend([
'-DBUILD_SHARED_LIBS=%s' % on_off('+shared')
])
if '+rocm' in spec:
args.append(
'-DHIP_ROOT_DIR={0}'.format(spec['hip'].prefix))
rocm_archs = spec.variants['amdgpu_target'].value
if 'none' not in rocm_archs:
args.append('-DHIP_HIPCC_FLAGS=--amdgpu-target={0}'.
format(",".join(rocm_archs)))
return args