163 lines
6.4 KiB
Python
163 lines
6.4 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 itertools
|
|
import re
|
|
|
|
from spack.package import *
|
|
|
|
|
|
class Rocsolver(CMakePackage):
|
|
"""rocSOLVER is a work-in-progress implementation of a
|
|
subset of LAPACK functionality on the ROCm platform."""
|
|
|
|
homepage = "https://github.com/ROCm/rocSOLVER"
|
|
git = "https://github.com/ROCm/rocSOLVER.git"
|
|
url = "https://github.com/ROCm/rocSOLVER/archive/rocm-6.1.2.tar.gz"
|
|
tags = ["rocm"]
|
|
|
|
maintainers("cgmb", "srekolam", "renjithravindrankannath", "haampie")
|
|
libraries = ["librocsolver"]
|
|
|
|
amdgpu_targets = ROCmPackage.amdgpu_targets
|
|
|
|
variant(
|
|
"amdgpu_target",
|
|
description="AMD GPU architecture",
|
|
values=auto_or_any_combination_of(*amdgpu_targets),
|
|
sticky=True,
|
|
)
|
|
variant(
|
|
"optimal",
|
|
default=True,
|
|
description="This option improves performance at the cost of increased binary \
|
|
size and compile time by adding specialized kernels \
|
|
for small matrix sizes",
|
|
)
|
|
variant("asan", default=False, description="Build with address-sanitizer enabled or disabled")
|
|
|
|
conflicts("+asan", when="os=rhel9")
|
|
conflicts("+asan", when="os=centos7")
|
|
conflicts("+asan", when="os=centos8")
|
|
|
|
license("BSD-2-Clause")
|
|
|
|
version("develop", branch="develop")
|
|
version("master", branch="master")
|
|
version("6.2.1", sha256="e1c19cd25f7119c116d1c63e2384e9c47a4ff7ae14bb42dfcef766a4d3a011d5")
|
|
version("6.2.0", sha256="74cb799dcddfcbd6ee05398003416dbccd3d06d7f4b23e4324baac3f15440162")
|
|
version("6.1.2", sha256="8cb45b6a4ed819b8e952c0bfdd8bf7dd941478ac656bea42a6d6751f459e66ea")
|
|
version("6.1.1", sha256="3bbba30fa7f187676caf858f66c2345e4dcc81b9546eca4a726c0b159dad22bd")
|
|
version("6.1.0", sha256="f1d7a4edf14ed0b2e2f74aa5cbc9db0c3b0dd31e50bbada1586cb353a28fe015")
|
|
version("6.0.2", sha256="781d5df2886ab0d5087a215a33ac390dd27653b2a9b4a620c7d51b0ae56f63d2")
|
|
version("6.0.0", sha256="5fcaba96f3efafc2ecc3f4ec104095d96545c16e1b9f95410bd571cb0fc643ae")
|
|
version("5.7.1", sha256="83e0c137b8690dbeb2e85d9e25415d96bd06979f09f2b10b2aff8e4c9f833fa4")
|
|
version("5.7.0", sha256="bb16d360f14b34fe6e8a6b8ddc6e631672a5ffccbdcb25f0ce319edddd7f9682")
|
|
version("5.6.1", sha256="6a8f366218aee599a0e56755030f94ee690b34f30e6d602748632226c5dc21bb")
|
|
version("5.6.0", sha256="54baa7f35f3c53da9005054e6f7aeecece5526dafcb277af32cbcb3996b0cbbc")
|
|
version("5.5.1", sha256="8bf843e42d2e89203ea5fdb6e6082cea90da8d02920ab4c09bcc2b6f69909760")
|
|
version("5.5.0", sha256="6775aa5b96731208c12c5b450cf218d4c262a80b7ea20c2c3034c448bb2ca4d2")
|
|
with default_args(deprecated=True):
|
|
version("5.4.3", sha256="5308b68ea72f465239a4bb2ed1a0507f0df7c98d3df3fd1f392e6d9ed7975232")
|
|
version("5.4.0", sha256="69690839cb649dee43353b739d3e6b2312f3d965dfe66705c0ea910e57c6a8cb")
|
|
version("5.3.3", sha256="d2248b5e2e0b20e08dd1ee5408e38deb02ecd28096dc7c7f2539351df6cb6ad5")
|
|
version("5.3.0", sha256="4569f860d240d50e94e77d498050f5cafe5ad11daddaead3e7e9eaa1957878a7")
|
|
|
|
depends_on("c", type="build") # generated
|
|
depends_on("cxx", type="build") # generated
|
|
|
|
depends_on("cmake@3.8:", type="build")
|
|
depends_on("fmt@7:", type="build", when="@4.5.0:")
|
|
depends_on("fmt@7:8.0.1", type="test", when="@5.6:")
|
|
|
|
depends_on("googletest@1.10.0:", type="test")
|
|
depends_on("netlib-lapack@3.7.1:", type="test")
|
|
|
|
# Maximize compatibility with other libraries that are using fmt.
|
|
patch("fmt-9-compatibility.patch", when="@5.2.0:5.5")
|
|
|
|
depends_on("hip")
|
|
depends_on("rocm-cmake@master", type="build", when="@master:")
|
|
depends_on("rocm-cmake@4.5.0:", type="build", when="@4.5.0:")
|
|
depends_on("rocm-cmake@4.3.0:", type="build", when="@4.3.0:")
|
|
depends_on("rocm-cmake@3.5.0:", type="build")
|
|
depends_on("rocsparse@5.2:", when="@5.6:")
|
|
|
|
for ver in ["master", "develop"]:
|
|
depends_on(f"rocblas@{ver}", when=f"@{ver}")
|
|
|
|
for ver in [
|
|
"5.3.0",
|
|
"5.3.3",
|
|
"5.4.0",
|
|
"5.4.3",
|
|
"5.5.0",
|
|
"5.5.1",
|
|
"5.6.0",
|
|
"5.6.1",
|
|
"5.7.0",
|
|
"5.7.1",
|
|
"6.0.0",
|
|
"6.0.2",
|
|
"6.1.0",
|
|
"6.1.1",
|
|
"6.1.2",
|
|
"6.2.0",
|
|
"6.2.1",
|
|
]:
|
|
depends_on(f"hip@{ver}", when=f"@{ver}")
|
|
depends_on(f"rocblas@{ver}", when=f"@{ver}")
|
|
|
|
for tgt in itertools.chain(["auto"], amdgpu_targets):
|
|
depends_on(f"rocblas amdgpu_target={tgt}", when=f"amdgpu_target={tgt}")
|
|
|
|
@classmethod
|
|
def determine_version(cls, lib):
|
|
match = re.search(r"lib\S*\.so\.\d+\.\d+\.(\d)(\d\d)(\d\d)", lib)
|
|
if match:
|
|
return "{0}.{1}.{2}".format(
|
|
int(match.group(1)), int(match.group(2)), int(match.group(3))
|
|
)
|
|
return None
|
|
|
|
def cmake_args(self):
|
|
args = [
|
|
self.define("BUILD_CLIENTS_SAMPLES", "OFF"),
|
|
self.define("BUILD_CLIENTS_TESTS", self.run_tests),
|
|
self.define("BUILD_CLIENTS_BENCHMARKS", "OFF"),
|
|
self.define_from_variant("OPTIMAL", "optimal"),
|
|
self.define("ROCSOLVER_EMBED_FMT", "ON"),
|
|
]
|
|
|
|
tgt = self.spec.variants["amdgpu_target"]
|
|
if "auto" not in tgt:
|
|
args.append(self.define_from_variant("AMDGPU_TARGETS", "amdgpu_target"))
|
|
|
|
if self.spec.satisfies("^cmake@3.21.0:3.21.2"):
|
|
args.append(self.define("__skip_rocmclang", "ON"))
|
|
|
|
if self.spec.satisfies("@5.2.0:"):
|
|
args.append(self.define("BUILD_FILE_REORG_BACKWARD_COMPATIBILITY", True))
|
|
if self.spec.satisfies("@5.3.0:"):
|
|
args.append("-DCMAKE_INSTALL_LIBDIR=lib")
|
|
|
|
return args
|
|
|
|
def setup_build_environment(self, env):
|
|
env.set("CXX", self.spec["hip"].hipcc)
|
|
if self.spec.satisfies("+asan"):
|
|
env.set("CC", f"{self.spec['llvm-amdgpu'].prefix}/bin/clang")
|
|
env.set("CXX", f"{self.spec['llvm-amdgpu'].prefix}/bin/clang++")
|
|
env.set("ASAN_OPTIONS", "detect_leaks=0")
|
|
env.set("CFLAGS", "-fsanitize=address -shared-libasan")
|
|
env.set("CXXFLAGS", "-fsanitize=address -shared-libasan")
|
|
env.set("LDFLAGS", "-fuse-ld=lld")
|
|
|
|
@run_after("build")
|
|
@on_package_attributes(run_tests=True)
|
|
def check_build(self):
|
|
exe = Executable(join_path(self.build_directory, "clients", "staging", "rocsolver-test"))
|
|
exe("--gtest_filter=checkin*-*known_bug*")
|