From 2712ea6299f34d8138254ff95559cb18f78989c0 Mon Sep 17 00:00:00 2001 From: fpruvost Date: Wed, 25 May 2022 19:15:53 +0200 Subject: [PATCH] Pastix: new package (#30533) --- .../repos/builtin/packages/pastix/package.py | 88 +++++++++++++++++++ 1 file changed, 88 insertions(+) create mode 100644 var/spack/repos/builtin/packages/pastix/package.py diff --git a/var/spack/repos/builtin/packages/pastix/package.py b/var/spack/repos/builtin/packages/pastix/package.py new file mode 100644 index 0000000000..63ad654b93 --- /dev/null +++ b/var/spack/repos/builtin/packages/pastix/package.py @@ -0,0 +1,88 @@ +# Copyright 2013-2022 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) + + +class Pastix(CMakePackage, CudaPackage): + """a high performance parallel solver for very large sparse linear systems + based on direct methods""" + homepage = "https://gitlab.inria.fr/solverstack/pastix/blob/master/README.md" + url = "https://gitlab.inria.fr/solverstack/pastix/uploads/baa57033d98378e0f3affbf45900fb6e/pastix-6.2.1.tar.gz" + git = "https://gitlab.inria.fr/solverstack/pastix.git" + maintainers = ['fpruvost', 'mfaverge', 'ramet'] + + version('master', branch='master', submodules=True) + version('6.2.1', '50742cc0e6e03728f7529fc607d5c65b9e14205f192947678d6103a872a6210c') + + # cmake's specific + variant('shared', default=True, description='Build Pastix as a shared library') + + # pastix's specific + variant('doc', default=False, description='Enable documentation') + variant('int64', default=False, description='To use 64 bits integers') + variant('metis', default=False, description='Enable Metis') + variant('scotch', default=True, description='Enable Scotch') + variant( + 'runtime', default='none', description='Runtime support', + values=('none', 'starpu'), multi=False + ) + variant('cuda', default=False, when='runtime=starpu', description='Enable CUDA') + variant('mpi', default=False, description='Enable MPI') + + # Dependencies + depends_on("pkgconfig", type='build') + depends_on("hwloc") + depends_on("lapack") + # ensure openblas use threads=openmp to be thread-safe + depends_on('openblas threads=openmp', when='^openblas') + with when("+metis"): + depends_on("metis@5.1:") + depends_on("metis@5.1:+int64", when='+int64') + with when("+scotch"): + depends_on("scotch~metis") + depends_on("scotch~metis+int64", when='+int64') + depends_on("scotch~metis~mpi", when='~mpi') + with when("runtime=starpu"): + depends_on("starpu") + depends_on("starpu~mpi", when='~mpi') + depends_on("starpu+mpi", when='+mpi') + depends_on("starpu~cuda", when='~cuda') + depends_on("starpu+cuda", when='+cuda') + depends_on("cuda", when='+cuda') + depends_on("mpi", when='+mpi') + + def cmake_args(self): + spec = self.spec + + args = [ + "-Wno-dev", + self.define("CMAKE_COLOR_MAKEFILE", "ON"), + self.define("CMAKE_VERBOSE_MAKEFILE", "ON"), + self.define_from_variant("BUILD_SHARED_LIBS", "shared"), + self.define_from_variant("PASTIX_INT64", "int64"), + self.define_from_variant("PASTIX_ORDERING_METIS", "metis"), + self.define_from_variant("PASTIX_ORDERING_SCOTCH", "scotch"), + self.define("PASTIX_WITH_PARSEC", "OFF"), + self.define_from_variant("PASTIX_WITH_MPI", "mpi") + ] + + if spec.satisfies('runtime=starpu'): + args.extend([self.define("PASTIX_WITH_STARPU", "ON")]) + args.extend([self.define_from_variant("PASTIX_WITH_CUDA", "cuda")]) + + if ('^intel-mkl' in spec or '^intel-parallel-studio+mkl' in spec): + args.extend([self.define("BLA_VENDOR", "Intel10_64lp_seq")]) + elif ('^netlib-lapack' in spec): + args.extend([self.define("BLA_VENDOR", "Generic")]) + elif ('^openblas' in spec): + args.extend([self.define("BLA_VENDOR", "OpenBLAS")]) + + if spec.satisfies('+mpi'): + args.extend([ + self.define("MPI_C_COMPILER", self.spec['mpi'].mpicc), + self.define("MPI_CXX_COMPILER", self.spec['mpi'].mpicxx), + self.define("MPI_Fortran_COMPILER", self.spec['mpi'].mpifc) + ]) + + return args