Update hiop package (#22232)
Co-authored-by: Harmen Stoppels <harmenstoppels@gmail.com>
This commit is contained in:
Asher Mancinelli
GitHub
parent
a63a3c1d3d
commit
e3bcb0ec1e
@ -6,7 +6,7 @@
|
|||||||
from spack import *
|
from spack import *
|
||||||
|
|
||||||
|
|
||||||
class Hiop(CMakePackage):
|
class Hiop(CMakePackage, CudaPackage):
|
||||||
"""HiOp is an optimization solver for solving certain mathematical
|
"""HiOp is an optimization solver for solving certain mathematical
|
||||||
optimization problems expressed as nonlinear programming problems.
|
optimization problems expressed as nonlinear programming problems.
|
||||||
HiOp is a lightweight HPC solver that leverages application's existing
|
HiOp is a lightweight HPC solver that leverages application's existing
|
||||||
@ -16,18 +16,54 @@ class Hiop(CMakePackage):
|
|||||||
homepage = "https://github.com/LLNL/hiop"
|
homepage = "https://github.com/LLNL/hiop"
|
||||||
git = "https://github.com/LLNL/hiop.git"
|
git = "https://github.com/LLNL/hiop.git"
|
||||||
|
|
||||||
|
# Most recent tagged snapshot is the preferred version when profiling.
|
||||||
|
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')
|
version('0.1', tag='v0.1')
|
||||||
|
|
||||||
variant('mpi', default=True,
|
# Development branches
|
||||||
description='Enable/Disable MPI')
|
version('master', branch='master')
|
||||||
|
version('develop', branch='develop')
|
||||||
|
|
||||||
variant('deepchecking', default=True,
|
variant(
|
||||||
description='Ultra safety checks - \
|
'jsrun',
|
||||||
used for increased robustness and self-diagnostics')
|
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('mpi', when='+mpi')
|
|
||||||
depends_on('lapack')
|
depends_on('lapack')
|
||||||
depends_on('blas')
|
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
|
flag_handler = build_system_flags
|
||||||
|
|
||||||
@ -35,16 +71,6 @@ def cmake_args(self):
|
|||||||
args = []
|
args = []
|
||||||
spec = self.spec
|
spec = self.spec
|
||||||
|
|
||||||
if '+mpi' in spec:
|
|
||||||
args.append("-DWITH_MPI=ON")
|
|
||||||
else:
|
|
||||||
args.append("-DWITH_MPI=OFF")
|
|
||||||
|
|
||||||
if '+deepchecking' in spec:
|
|
||||||
args.append("-DDEEP_CHECKING=ON")
|
|
||||||
else:
|
|
||||||
args.append("-DDEEP_CHECKING=OFF")
|
|
||||||
|
|
||||||
lapack_blas_libs = (
|
lapack_blas_libs = (
|
||||||
spec['lapack'].libs + spec['blas'].libs).joined(';')
|
spec['lapack'].libs + spec['blas'].libs).joined(';')
|
||||||
args.extend([
|
args.extend([
|
||||||
@ -52,4 +78,84 @@ def cmake_args(self):
|
|||||||
'-DLAPACK_LIBRARIES={0}'.format(lapack_blas_libs)
|
'-DLAPACK_LIBRARIES={0}'.format(lapack_blas_libs)
|
||||||
])
|
])
|
||||||
|
|
||||||
|
args.append('-DHIOP_BUILD_STATIC=ON')
|
||||||
|
if '+shared' in spec:
|
||||||
|
args.append('-DHIOP_BUILD_SHARED=ON')
|
||||||
|
else:
|
||||||
|
args.append('-DHIOP_BUILD_SHARED=OFF')
|
||||||
|
|
||||||
|
if '+mpi' in spec:
|
||||||
|
args.append(
|
||||||
|
"-DHIOP_USE_MPI=ON -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))
|
||||||
|
else:
|
||||||
|
args.append("-DHIOP_USE_MPI=OFF")
|
||||||
|
|
||||||
|
if '+deepchecking' in spec:
|
||||||
|
args.append("-DHIOP_DEEPCHECKS=ON")
|
||||||
|
else:
|
||||||
|
args.append("-DHIOP_DEEPCHECKS=OFF")
|
||||||
|
|
||||||
|
# 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:
|
||||||
|
args.append("-DHIOP_USE_GPU=ON")
|
||||||
|
|
||||||
|
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))
|
||||||
|
args.append("-DHIOP_USE_CUDA=ON")
|
||||||
|
|
||||||
|
args.append('-DHIOP_USE_MAGMA=ON')
|
||||||
|
if '+magma' in spec:
|
||||||
|
args.append(
|
||||||
|
"-DHIOP_MAGMA_DIR={0}".format(spec['magma'].prefix))
|
||||||
|
|
||||||
|
else:
|
||||||
|
args.append("-DHIOP_USE_GPU=OFF")
|
||||||
|
args.append("-DHIOP_USE_CUDA=OFF")
|
||||||
|
args.append("-DHIOP_USE_MAGMA=OFF")
|
||||||
|
|
||||||
|
if '+raja' in spec:
|
||||||
|
args.append("-DHIOP_USE_RAJA=ON")
|
||||||
|
args.append("-DHIOP_USE_UMPIRE=ON")
|
||||||
|
args.append("-Dumpire_DIR={0}".format(spec['umpire'].prefix))
|
||||||
|
args.append("-DRAJA_DIR={0}".format(spec['raja'].prefix))
|
||||||
|
else:
|
||||||
|
args.append("-DHIOP_USE_RAJA=OFF")
|
||||||
|
args.append("-DHIOP_USE_UMPIRE=OFF")
|
||||||
|
|
||||||
|
if '+kron' in spec:
|
||||||
|
args.append("-DHIOP_WITH_KRON_REDUCTION=ON")
|
||||||
|
args.append(
|
||||||
|
"-DHIOP_UMFPACK_DIR={0}".format(spec['suite-sparse'].prefix))
|
||||||
|
else:
|
||||||
|
args.append("-DHIOP_WITH_KRON_REDUCTION=OFF")
|
||||||
|
|
||||||
|
# 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_SPARSE=ON")
|
||||||
|
args.append("-DHIOP_USE_COINHSL=ON")
|
||||||
|
args.append(
|
||||||
|
"-DHIOP_COINHSL_DIR={0}".format(spec['coinhsl'].prefix))
|
||||||
|
else:
|
||||||
|
args.append("-DHIOP_SPARSE=OFF")
|
||||||
|
args.append("-DHIOP_USE_COINHSL=OFF")
|
||||||
|
|
||||||
|
# Enable CTest tests to use jsrun for easier testing on IBM systems
|
||||||
|
if '+jsrun' in spec:
|
||||||
|
args.append("-DHIOP_TEST_WITH_BSUB=ON")
|
||||||
|
else:
|
||||||
|
args.append("-DHIOP_TEST_WITH_BSUB=OFF")
|
||||||
|
|
||||||
return args
|
return args
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user