zhangyiss/spack
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

spla: add v1.6.0 (#44609)

Browse Source
This commit is contained in:
Simon Frasch 2024-06-11 13:06:50 +02:00 committed by GitHub
parent 0781615117
commit 82a932c078
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 28 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
var/spack/repos/builtin/packages/spla/package.py
View File

@ -19,6 +19,7 @@ class Spla(CMakePackage):
license("BSD-3-Clause")
version("1.6.0", sha256="917c24e2a768499967eba47b2cc2475df9fabee327b7821d24970b6a08055c09")
version("1.5.5", sha256="bc0c366e228344b1b2df55b9ce750d73c1165380e512da5a04d471db126d66ce")
version("1.5.4", sha256="de30e427d24c741e2e4fcae3d7668162056ac2574afed6522c0bb49d6f1d0f79")
version("1.5.3", sha256="527c06e316ce46ec87309a16bfa4138b1abad23fd276fe789c78a2de84f05637")
@ -35,7 +36,7 @@ class Spla(CMakePackage):
version("develop", branch="develop")
version("master", branch="master")
variant("openmp", default=True, description="Build with OpenMP support")
variant("openmp", default=True, when="@:1.5.5", description="Build with OpenMP support")
variant("static", default=False, description="Build as static library")
variant("cuda", default=False, description="CUDA backend")
variant("rocm", default=False, description="ROCm backend")
@ -51,10 +52,14 @@ class Spla(CMakePackage):
depends_on("mpi")
depends_on("blas")
depends_on("cmake@3.10:", type="build")
depends_on("cmake@3.18:", type="build", when="@1.6.0:")
depends_on("cuda", when="+cuda")
depends_on("rocblas", when="+rocm")
depends_on("cuda@11:", when="@1.6.0: +cuda")
depends_on("hip", when="+rocm")
depends_on("rocblas", when="+rocm")
conflicts("^rocblas@6.0.0:", when="@:1.5.5 +rocm")
# Propagate openmp to blas
depends_on("openblas threads=openmp", when="+openmp ^[virtuals=blas] openblas")
@ -67,30 +72,36 @@ class Spla(CMakePackage):
patch("0001-amd_blis.patch", when="@1.3.0:1.4.0 ^amdblis")
def cmake_args(self):
spec = self.spec
args = [
self.define_from_variant("SPLA_OMP", "openmp"),
self.define_from_variant("SPLA_FORTRAN", "fortran"),
self.define_from_variant("SPLA_STATIC", "static"),
]
if "+cuda" in self.spec:
if "+cuda" in spec:
args += ["-DSPLA_GPU_BACKEND=CUDA"]
elif "+rocm" in self.spec:
elif "+rocm" in spec:
args += ["-DSPLA_GPU_BACKEND=ROCM"]
else:
args += ["-DSPLA_GPU_BACKEND=OFF"]
if self.spec["blas"].name == "openblas":
# v1.6.0: No longer has custom BLAS detection and only uses the FindBLAS CMake module.
if spec.satisfies("@:1.5.5"):
args += [self.define_from_variant("SPLA_OMP", "openmp")]
if spec["blas"].name == "openblas":
args += ["-DSPLA_HOST_BLAS=OPENBLAS"]
elif self.spec["blas"].name in ["amdblis", "blis"]:
elif spec["blas"].name in ["amdblis", "blis"]:
args += ["-DSPLA_HOST_BLAS=BLIS"]
elif self.spec["blas"].name == "atlas":
elif spec["blas"].name == "atlas":
args += ["-DSPLA_HOST_BLAS=ATLAS"]
elif self.spec["blas"].name == "intel-mkl":
elif spec["blas"].name == "intel-mkl":
args += ["-DSPLA_HOST_BLAS=MKL"]
elif self.spec["blas"].name == "netlib-lapack":
elif spec["blas"].name == "netlib-lapack":
args += ["-DSPLA_HOST_BLAS=GENERIC"]
elif self.spec["blas"].name == "cray-libsci":
elif spec["blas"].name == "cray-libsci":
args += ["-DSPLA_HOST_BLAS=CRAY_LIBSCI"]
else:
args += [self.define("BLAS_LIBRARIES", spec["blas"].libs.joined(";"))]
return args