4d3b85c4d4
Add various missing imports in packages. Remove redundant imports Export NoLibrariesError, NoHeadersError, which_string in spack.package
206 lines
8.2 KiB
Python
206 lines
8.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)
|
|
|
|
import os
|
|
|
|
import spack.version
|
|
from spack.package import *
|
|
|
|
|
|
class SingularityEos(CMakePackage, CudaPackage):
|
|
"""Singularity-EOS: A collection of closure models and tools useful for
|
|
multiphysics codes."""
|
|
|
|
homepage = "https://lanl.github.io/singularity-eos/main/index.html"
|
|
git = "https://github.com/lanl/singularity-eos.git"
|
|
url = "https://github.com/lanl/singularity-eos/archive/refs/tags/release-1.6.1.tar.gz"
|
|
|
|
maintainers("rbberger")
|
|
|
|
license("BSD-3-Clause")
|
|
|
|
version("main", branch="main")
|
|
version("1.8.0", sha256="1f1ec496f714aa23cc7003c88a85bd10d0e53e37659ba7310541248e48a66558")
|
|
version("1.7.0", sha256="ce0825db2e9d079503e98cecf1c565352be696109042b3a0941762b35f36dc49")
|
|
version(
|
|
"1.6.2",
|
|
sha256="9c85fca679139a40cc9c72fcaeeca78a407cc1ca184734785236042de364b942",
|
|
deprecated=True,
|
|
)
|
|
version(
|
|
"1.6.1",
|
|
sha256="c6d92dfecf9689ffe2df615791c039f7e527e9f47799a862e26fa4e3420fe5d7",
|
|
deprecated=True,
|
|
)
|
|
|
|
# build with kokkos, kokkos-kernels for offloading support
|
|
variant("kokkos", default=False, description="Enable kokkos")
|
|
variant(
|
|
"kokkos-kernels", default=False, description="Enable kokkos-kernals for linear algebra"
|
|
)
|
|
|
|
variant("openmp", default=False, description="Enable openmp")
|
|
|
|
# for compatibility with downstream projects
|
|
variant("mpi", default=False, description="Build with MPI support")
|
|
|
|
# build converters for sesame, stellarcollapse eos's
|
|
variant(
|
|
"build_extra",
|
|
description="Build converters",
|
|
values=any_combination_of("sesame", "stellarcollapse").with_default("none"),
|
|
)
|
|
|
|
# build the Fortran interface
|
|
variant("fortran", default=True, description="Enable building fortran interface")
|
|
|
|
# build the Python bindings
|
|
variant("python", default=False, description="Enable building Python bindings")
|
|
|
|
variant("eospac", default=True, description="Enable EOSPAC for table reads")
|
|
|
|
variant("hdf5", default=False, description="Enable HDF5 support")
|
|
|
|
variant("spiner", default=True, description="Use Spiner")
|
|
|
|
variant("closure", default=True, description="Build closure module")
|
|
|
|
depends_on("c", type="build")
|
|
depends_on("cxx", type="build")
|
|
depends_on("fortran", type="build", when="+fortran")
|
|
|
|
# building/testing/docs
|
|
depends_on("cmake@3.19:", type="build")
|
|
depends_on("python@3:", when="+python")
|
|
depends_on("py-pybind11@2.9.1:", when="+python")
|
|
depends_on("catch2@2.13.7", type="test")
|
|
depends_on("py-numpy", type="test")
|
|
|
|
# linear algebra when not using GPUs
|
|
depends_on("eigen@3.3.8", when="~kokkos-kernels~cuda")
|
|
|
|
depends_on("eospac", when="+eospac")
|
|
|
|
depends_on("ports-of-call@1.4.2,1.5.2:", when="@:1.7.0")
|
|
depends_on("ports-of-call@1.5.2:", when="@1.7.1:")
|
|
depends_on("ports-of-call@main", when="@main")
|
|
|
|
depends_on("spiner +kokkos", when="+kokkos+spiner")
|
|
depends_on("spiner +hdf5", when="+hdf5+spiner")
|
|
|
|
depends_on("spiner@:1.6.0", when="@:1.7.0 +spiner")
|
|
depends_on("spiner@1.6.1:", when="@1.8.0: +spiner")
|
|
depends_on("spiner@main", when="@main +spiner")
|
|
|
|
depends_on("mpark-variant")
|
|
depends_on(
|
|
"mpark-variant",
|
|
patches=patch(
|
|
"https://raw.githubusercontent.com/lanl/singularity-eos/main/utils/cuda_compatibility.patch",
|
|
sha256="7b3eaa52b5ab23dc45fbfb456528e36742e04b838a5df859eca96c4e8274bb38",
|
|
),
|
|
when="+cuda",
|
|
)
|
|
|
|
for _myver, _kver in zip(("@:1.6.2", "@1.7.0:"), ("@3.2:", "@3.3:")):
|
|
depends_on("kokkos" + _kver, when=_myver + "+kokkos")
|
|
depends_on("kokkos-kernels" + _kver, when=_myver + "+kokkos-kernels")
|
|
|
|
# set up kokkos offloading dependencies
|
|
for _flag in ("~cuda", "+cuda", "~openmp", "+openmp"):
|
|
depends_on("kokkos ~shared" + _flag, when="+kokkos" + _flag)
|
|
depends_on("kokkos-kernels" + _flag, when="+kokkos-kernels" + _flag)
|
|
depends_on("spiner" + _flag, when="+kokkos" + _flag)
|
|
|
|
# specfic specs when using GPU/cuda offloading
|
|
depends_on("kokkos +wrapper+cuda_lambda", when="+cuda+kokkos")
|
|
|
|
# fix for older spacks
|
|
if spack.version.Version(spack.spack_version) >= spack.version.Version("0.17"):
|
|
depends_on("kokkos-kernels ~shared", when="+kokkos-kernels")
|
|
|
|
for _flag in list(CudaPackage.cuda_arch_values):
|
|
depends_on("kokkos cuda_arch=" + _flag, when="+cuda+kokkos cuda_arch=" + _flag)
|
|
depends_on("kokkos-kernels cuda_arch=" + _flag, when="+cuda+kokkos cuda_arch=" + _flag)
|
|
depends_on("spiner cuda_arch=" + _flag, when="+cuda+kokkos cuda_arch=" + _flag)
|
|
|
|
conflicts("cuda_arch=none", when="+cuda", msg="CUDA architecture is required")
|
|
|
|
# these are mirrored in the cmake configuration
|
|
conflicts("+cuda", when="~kokkos")
|
|
conflicts("+openmp", when="~kokkos")
|
|
conflicts("+kokkos-kernels", when="~kokkos")
|
|
conflicts("+hdf5", when="~spiner")
|
|
|
|
conflicts("+fortran", when="~closure")
|
|
|
|
# NOTE: these are set so that dependencies in downstream projects share
|
|
# common MPI dependence
|
|
for _flag in ("~mpi", "+mpi"):
|
|
depends_on("hdf5~cxx+hl" + _flag, when="+hdf5" + _flag)
|
|
depends_on("py-h5py" + _flag, when="@:1.6.2 " + _flag)
|
|
|
|
# can be removed once <1.8.0 versions have been removed
|
|
def flag_handler(self, name, flags):
|
|
if name == "fflags":
|
|
if self.spec.satisfies("%cce+fortran"):
|
|
# The Cray fortran compiler generates module files with
|
|
# uppercase names by default, which is not handled by the
|
|
# CMake scripts. The following flag forces the compiler to
|
|
# produce module files with lowercase names.
|
|
flags.append("-ef")
|
|
return (flags, None, None)
|
|
|
|
def cmake_args(self):
|
|
args = [
|
|
self.define("SINGULARITY_PATCH_MPARK_VARIANT", False),
|
|
self.define_from_variant("SINGULARITY_USE_CUDA", "cuda"),
|
|
self.define_from_variant("SINGULARITY_USE_KOKKOS", "kokkos"),
|
|
self.define_from_variant("SINGULARITY_USE_KOKKOSKERNELS", "kokkos-kernels"),
|
|
self.define_from_variant("SINGULARITY_USE_FORTRAN", "fortran"),
|
|
self.define_from_variant("SINGULARITY_BUILD_CLOSURE", "closure"),
|
|
self.define_from_variant("SINGULARITY_BUILD_PYTHON", "python"),
|
|
self.define_from_variant("SINGULARITY_USE_SPINER", "spiner"),
|
|
self.define_from_variant("SINGULARITY_USE_SPINER_WITH_HDF5", "hdf5"),
|
|
self.define("SINGULARITY_BUILD_TESTS", self.run_tests),
|
|
self.define(
|
|
"SINGULARITY_BUILD_SESAME2SPINER",
|
|
"sesame" in self.spec.variants["build_extra"].value,
|
|
),
|
|
self.define(
|
|
"SINGULARITY_TEST_SESAME",
|
|
("sesame" in self.spec.variants["build_extra"].value and self.run_tests),
|
|
),
|
|
self.define(
|
|
"SINGULARITY_BUILD_STELLARCOLLAPSE2SPINER",
|
|
"stellarcollapse" in self.spec.variants["build_extra"].value,
|
|
),
|
|
self.define(
|
|
"SINGULARITY_TEST_STELLARCOLLAPSE2SPINER",
|
|
("stellarcollapse" in self.spec.variants["build_extra"].value and self.run_tests),
|
|
),
|
|
self.define("SINGULARITY_TEST_PYTHON", ("+python" in self.spec and self.run_tests)),
|
|
self.define_from_variant("SINGULARITY_USE_HDF5", "hdf5"),
|
|
self.define_from_variant("SINGULARITY_USE_EOSPAC", "eospac"),
|
|
]
|
|
|
|
if "+kokkos+cuda" in self.spec:
|
|
args.append(self.define("CMAKE_CXX_COMPILER", self.spec["kokkos"].kokkos_cxx))
|
|
|
|
return args
|
|
|
|
def setup_run_environment(self, env):
|
|
if os.path.isdir(self.prefix.lib64):
|
|
lib_dir = self.prefix.lib64
|
|
else:
|
|
lib_dir = self.prefix.lib
|
|
|
|
if "+python" in self.spec:
|
|
python_version = self.spec["python"].version.up_to(2)
|
|
python_inst_dir = join_path(
|
|
lib_dir, "python{0}".format(python_version), "site-packages"
|
|
)
|
|
env.prepend_path("PYTHONPATH", python_inst_dir)
|