diff options
Diffstat (limited to 'var/spack/repos/builtin/packages/hiop/package.py')
| -rw-r--r-- | var/spack/repos/builtin/packages/hiop/package.py | 112 |
1 files changed, 93 insertions, 19 deletions
diff --git a/var/spack/repos/builtin/packages/hiop/package.py b/var/spack/repos/builtin/packages/hiop/package.py index f0603381dd..4751805e93 100644 --- a/var/spack/repos/builtin/packages/hiop/package.py +++ b/var/spack/repos/builtin/packages/hiop/package.py @@ -1,4 +1,4 @@ -# Copyright 2013-2020 Lawrence Livermore National Security, LLC and other +# 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) @@ -6,7 +6,7 @@ from spack import * -class Hiop(CMakePackage): +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 @@ -14,20 +14,58 @@ class Hiop(CMakePackage): specialized linear algebra kernels.""" 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.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') - variant('mpi', default=True, - description='Enable/Disable MPI') + # Development branches + version('master', branch='master') + version('develop', branch='develop') - variant('deepchecking', default=True, - description='Ultra safety checks - \ - used for increased robustness and self-diagnostics') + 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('mpi', when='+mpi') 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 @@ -35,16 +73,6 @@ class Hiop(CMakePackage): args = [] 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 = ( spec['lapack'].libs + spec['blas'].libs).joined(';') args.extend([ @@ -52,4 +80,50 @@ class Hiop(CMakePackage): '-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 |