hlrs-software-stack/spack
7
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

MFEM: add logic to find CUDA math-libs when using HPC SDK installation (#40815)

Browse source 
* mfem: add logic to find CUDA math-libs when using HPC SDK installation

* [@spackbot] updating style on behalf of v-dobrev
This commit is contained in:
Veselin Dobrev 2023-11-02 20:19:11 -07:00 committed by GitHub
parent 864d47043c
commit 48a21970d1
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 16 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
var/spack/repos/builtin/packages/mfem/package.py
View file

@ -919,6 +919,22 @@ def find_optional_library(name, prefix):
"CUDA_CXX=%s" % join_path(spec["cuda"].prefix, "bin", "nvcc"),
"CUDA_ARCH=sm_%s" % cuda_arch,
]
# Check if we are using a CUDA installation where the math libs are
# in a separate directory:
culibs = ["libcusparse"]
cuda_libs = find_optional_library(culibs, spec["cuda"].prefix)
if not cuda_libs:
p0 = os.path.realpath(join_path(spec["cuda"].prefix, "bin", "nvcc"))
p0 = os.path.dirname(p0)
p1 = os.path.dirname(p0)
while p1 != p0:
cuda_libs = find_optional_library(culibs, join_path(p1, "math_libs"))
if cuda_libs:
break
p0, p1 = p1, os.path.dirname(p1)
if not cuda_libs:
raise InstallError("Required CUDA libraries not found: %s" % culibs)
options += ["CUDA_LIB=%s" % ld_flags_from_library_list(cuda_libs)]
if "+rocm" in spec:
amdgpu_target = ",".join(spec.variants["amdgpu_target"].value)