603569e321
* Style: black 23, skip magic trailing commas * isort should use same line length as black * Fix unused import * Update version of black used in CI * Update new packages * Update new packages
68 lines
3.2 KiB
Python
68 lines
3.2 KiB
Python
# Copyright 2013-2023 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 PyPyfr(PythonPackage, CudaPackage, ROCmPackage):
|
|
"""PyFR is an open-source Python based framework for solving
|
|
advection-diffusion type problems on streaming architectures
|
|
using the Flux Reconstruction approach of Huynh."""
|
|
|
|
homepage = "http://www.pyfr.org/"
|
|
pypi = "pyfr/pyfr-1.13.0.tar.gz"
|
|
git = "https://github.com/PyFR/PyFR/"
|
|
maintainers("MichaelLaufer")
|
|
|
|
# git branches
|
|
version("develop", branch="develop")
|
|
version("master", branch="master")
|
|
|
|
# pypi releases
|
|
version("1.15.0", sha256="6a634b9d32447f45d3c24c9de0ed620a0a0a781be7cc5e57b1c1bf44a4650d8d")
|
|
version("1.14.0", sha256="ebf40ce0896cce9ac802e03fd9430b5be30ea837c31224531a6d5fd68f820766")
|
|
version("1.13.0", sha256="ac6ecec738d4e23799ab8c50dea9bdbd7d37bc971bd33f22720c5a230b8e7b2f")
|
|
|
|
variant("metis", default=True, description="Metis for mesh partitioning")
|
|
variant("scotch", default=False, description="Scotch for mesh partitioning")
|
|
variant("cuda", default=False, description="CUDA backend support")
|
|
variant("hip", default=False, description="HIP backend support")
|
|
variant("libxsmm", default=True, description="LIBXSMM for OpenMP backend")
|
|
variant("scipy", default=True, description="Scipy acceleration for point sampling")
|
|
|
|
# Required dependencies
|
|
depends_on("python@3.9:", type=("build", "run"))
|
|
depends_on("py-setuptools", type="build")
|
|
depends_on("py-gimmik@2.3:2", when="@:1.14.0", type=("build", "run"))
|
|
depends_on("py-gimmik@3", when="@1.15.0:", type=("build", "run"))
|
|
depends_on("py-h5py@2.10:", type=("build", "run"))
|
|
depends_on("py-mako@1.0.0:", type=("build", "run"))
|
|
depends_on("py-mpi4py@3.1.0:", type=("build", "run"))
|
|
depends_on("py-numpy@1.20:+blas", type=("build", "run"))
|
|
depends_on("py-platformdirs@2.2.0:", type=("build", "run"))
|
|
depends_on("py-pytools@2016.2.1:", type=("build", "run"))
|
|
|
|
# Optional dependencies
|
|
depends_on("py-scipy", when="+scipy", type=("build", "run"))
|
|
depends_on("metis@5.0:", when="+metis", type=("run"))
|
|
depends_on("scotch@7.0.1: +link_error_lib", when="+scotch", type=("run"))
|
|
depends_on("cuda@8.0: +allow-unsupported-compilers", when="@:1.14.0 +cuda", type=("run"))
|
|
depends_on("cuda@11.4.0: +allow-unsupported-compilers", when="@1.15.0: +cuda", type=("run"))
|
|
depends_on("rocblas@5.2.0:", when="+hip", type=("run"))
|
|
depends_on("libxsmm@1.18:+shared blas=0", when="+libxsmm", type=("run"))
|
|
|
|
# Conflicts for compilers not supporting OpenMP 5.1+ from v1.15.0:
|
|
conflicts("%gcc@:11", when="@1.15.0: +libxsmm", msg="OpenMP 5.1+ supported compiler required!")
|
|
|
|
# Explicitly add dependencies to PYFR_LIBRARY_PATH environment variable
|
|
def setup_run_environment(self, env):
|
|
deps = ["metis", "scotch", "libxsmm", "cuda", "hip", "rocblas"]
|
|
pyfr_library_path = []
|
|
for dep in deps:
|
|
if "+{}".format(dep) in self.spec:
|
|
pyfr_library_path.extend(self.spec[dep].libs.directories)
|
|
env.set("PYFR_LIBRARY_PATH", ":".join(pyfr_library_path))
|