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

resolve: add package with cuda and rocm support (#40871)

Browse Source
This commit is contained in:
Cameron Rutherford 2023-12-01 21:49:11 -05:00 committed by GitHub
parent 1509e54435
commit 8bbc2e2ade
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 62 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

62
var/spack/repos/builtin/packages/resolve/package.py Normal file
View File

@ -0,0 +1,62 @@
# 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 Resolve(CMakePackage, CudaPackage, ROCmPackage):
"""ReSolve is a library of GPU-resident sparse linear solvers. It contains iterative and direct
solvers designed to run on NVIDIA and AMD GPUs, as well as CPU devices."""
homepage = "https://github.com/ORNL/ReSolve"
git = "https://github.com/ORNL/ReSolve.git"
maintainers("cameronrutherford", "pelesh", "ryandanehy", "kswirydo")
# version("1.0.0", submodules=False, branch="develop")
version("develop", submodules=False, branch="develop")
variant("klu", default=True, description="Use KLU, AMD and COLAMD Libraries from SuiteSparse")
depends_on("suite-sparse", when="+klu")
with when("+rocm"):
# Need at least 5.6+
depends_on("rocsparse@5.6:")
depends_on("rocblas@5.6:")
depends_on("rocsolver@5.6:")
# Optional profiling dependecies
# Will be controlled by variant in the future
# depends_on("roctracer-dev@5.6:")
# depends_on("roctracer-dev-api@5.6:")
# depends_on("rocprofiler-dev@5.6:")
def cmake_args(self):
args = []
spec = self.spec
args.extend(
[self.define("RESOLVE_USE_KLU", "klu"), self.define("RESOLVE_TEST_WITH_BSUB", False)]
)
if "+cuda" in spec:
cuda_arch_list = spec.variants["cuda_arch"].value
if cuda_arch_list[0] != "none":
args.append(self.define("CMAKE_CUDA_ARCHITECTURES", cuda_arch_list))
else:
args.append(self.define("CMAKE_CUDA_ARCHITECTURES", "70;75;80"))
args.append(self.define("RESOLVE_USE_CUDA", True))
elif "+rocm" in spec:
rocm_arch_list = spec.variants["amdgpu_target"].value
# `+rocm` conflicts with amdgpu_target == "none"...
# if rocm_arch_list[0] == "none":
# rocm_arch_list = "gfx90a"
args.append(self.define("GPU_TARGETS", rocm_arch_list))
args.append(self.define("AMDGPU_TARGETS", rocm_arch_list))
args.append(self.define("RESOLVE_USE_HIP", True))
return args