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
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
|
# 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 Hiop(CMakePackage, CudaPackage):
"""HiOp is an optimization solver for solving certain mathematical
optimization problems expressed as nonlinear programming problems.
HiOp is a lightweight HPC solver that leverages application's existing
data parallelism to parallelize the optimization iterations by using
specialized linear algebra kernels."""
homepage = "https://github.com/LLNL/hiop"
git = "https://github.com/LLNL/hiop.git"
# Most recent tagged snapshot is the preferred version when profiling.
version('0.4.1', tag='v0.4.1')
version('0.4', tag='v0.4')
version('0.3.99.3', tag='v0.3.99.3')
version('0.3.99.2', tag='v0.3.99.2')
version('0.3.99.1', tag='v0.3.99.1')
version('0.3.99.0', tag='v0.3.99.0')
version('0.3', tag='v0.3')
version('0.2', tag='v0.2')
version('0.1', tag='v0.1')
# Development branches
version('master', branch='master')
version('develop', branch='develop')
variant(
'jsrun',
default=False,
description='Enable/Disable jsrun command for testing')
variant(
'shared',
default=False,
description='Enable/Disable shared libraries')
variant('mpi', default=True, description='Enable/Disable MPI')
variant('raja', default=False, description='Enable/Disable RAJA')
variant('kron', default=False, description='Enable/Disable Kron reduction')
variant(
'sparse',
default=False,
description='Enable/Disable Sparse linear algebra')
variant('deepchecking', default=False,
description='Ultra safety checks - '
'used for increased robustness and self-diagnostics')
depends_on('lapack')
depends_on('blas')
depends_on('cmake@3.18:', type='build')
depends_on('mpi', when='+mpi')
depends_on('magma', when='+cuda')
depends_on('magma@2.5.4', when='@0.3.99.1:+cuda')
depends_on('raja', when='+raja')
depends_on('umpire', when='+raja')
depends_on('suite-sparse', when='+kron')
depends_on('coinhsl', when='+sparse')
depends_on('metis', when='+sparse')
flag_handler = build_system_flags
def cmake_args(self):
args = []
spec = self.spec
lapack_blas_libs = (
spec['lapack'].libs + spec['blas'].libs).joined(';')
args.extend([
'-DLAPACK_FOUND=TRUE',
'-DLAPACK_LIBRARIES={0}'.format(lapack_blas_libs)
])
args.append(self.define_from_variant('HIOP_BUILD_SHARED', 'shared'))
args.append(self.define_from_variant('HIOP_USE_MPI', 'mpi'))
args.append(self.define_from_variant('HIOP_DEEPCHECKS', 'deepchecking'))
args.append(self.define_from_variant('HIOP_USE_GPU', 'cuda'))
args.append(self.define_from_variant('HIOP_USE_CUDA', 'cuda'))
args.append(self.define_from_variant('HIOP_USE_MAGMA', 'cuda'))
args.append(self.define_from_variant('HIOP_USE_RAJA', 'raja'))
args.append(self.define_from_variant('HIOP_USE_UMPIRE', 'raja'))
args.append(self.define_from_variant('HIOP_WITH_KRON_REDUCTION', 'kron'))
args.append(self.define_from_variant('HIOP_SPARSE', 'sparse'))
args.append(self.define_from_variant('HIOP_USE_COINHSL', 'sparse'))
args.append(self.define_from_variant('HIOP_TEST_WITH_BSUB', 'jsrun'))
if '+mpi' in spec:
args.append('-DMPI_HOME={0}'.format(spec['mpi'].prefix))
args.append('-DMPI_C_COMPILER={0}'.format(spec['mpi'].mpicc))
args.append('-DMPI_CXX_COMPILER={0}'.format(spec['mpi'].mpicxx))
args.append('-DMPI_Fortran_COMPILER={0}'.format(spec['mpi'].mpifc))
# HIP flags are a part of the buildsystem, but full support is not
# yet ready for public release
args.append("-DHIOP_USE_HIP=OFF")
if '+cuda' in spec:
cuda_arch_list = spec.variants['cuda_arch'].value
cuda_arch = cuda_arch_list[0]
if cuda_arch != 'none':
args.append("-DHIOP_NVCC_ARCH=sm_{0}".format(cuda_arch))
args.append("-DCMAKE_CUDA_ARCHITECTURES={0}".format(cuda_arch))
if '+magma' in spec:
args.append(
"-DHIOP_MAGMA_DIR={0}".format(spec['magma'].prefix))
if '+kron' in spec:
args.append(
"-DHIOP_UMFPACK_DIR={0}".format(spec['suite-sparse'].prefix))
# Unconditionally disable strumpack, even when +sparse. This may be
# used in place of COINHSL for sparse interface, however this is not
# fully supported in spack at the moment.
args.append("-DHIOP_USE_STRUMPACK=OFF")
if '+sparse' in spec:
args.append(
"-DHIOP_COINHSL_DIR={0}".format(spec['coinhsl'].prefix))
return args
|
