102 lines
4.2 KiB
Python
102 lines
4.2 KiB
Python
# Copyright 2013-2024 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.package import *
|
|
|
|
|
|
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://files.inria.fr/pastix/releases/v6/pastix-6.3.2.tar.gz"
|
|
git = "https://gitlab.inria.fr/solverstack/pastix.git"
|
|
maintainers("fpruvost", "mfaverge", "ramet")
|
|
|
|
version("master", branch="master", submodules=True)
|
|
version("6.3.2", sha256="c4da8802d1933eecf8c09d7e63c014c81ccf353fe623142e9f5c5fc65ed82ee0")
|
|
version("6.3.1", sha256="290464d73b7d43356e4735a29932bf6f23a88e94ec7139ba7744c21e42c52681")
|
|
version("6.3.0", sha256="a6bfec32a3279d7b24c5fc05885c6632d177e467f1584707c6fd7c42a8703c3e")
|
|
version("6.2.2", sha256="cce9a1fe4678b5733c9f1a5a52f77b040eadc3e254418c6fb03d8ab37dede508")
|
|
version("6.2.1", sha256="b680cbfc265df8cba18d3a7093fcc02e260198c4a2d6a86d1e684bb291e309dd")
|
|
|
|
# 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="^[virtuals=lapack] 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
|