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

mfem: support petsc+rocm with spack-installed rocm (#40768)

Browse source
This commit is contained in:
Veselin Dobrev 2023-10-30 00:17:51 -07:00 committed by GitHub
parent 6983db1392
commit bd1bb7d1ba
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -923,6 +923,7 @@ def find_optional_library(name, prefix):
if "+rocm" in spec: if "+rocm" in spec:
amdgpu_target = ",".join(spec.variants["amdgpu_target"].value) amdgpu_target = ",".join(spec.variants["amdgpu_target"].value)
options += ["HIP_CXX=%s" % spec["hip"].hipcc, "HIP_ARCH=%s" % amdgpu_target] options += ["HIP_CXX=%s" % spec["hip"].hipcc, "HIP_ARCH=%s" % amdgpu_target]
hip_headers = HeaderList([])
hip_libs = LibraryList([]) hip_libs = LibraryList([])
# To use a C++ compiler that supports -xhip flag one can use # To use a C++ compiler that supports -xhip flag one can use
# something like this: # something like this:
@ -933,7 +934,7 @@ def find_optional_library(name, prefix):
# hip_libs += find_libraries("libamdhip64", spec["hip"].prefix.lib) # hip_libs += find_libraries("libamdhip64", spec["hip"].prefix.lib)
if "^hipsparse" in spec: # hipsparse is needed @4.4.0:+rocm if "^hipsparse" in spec: # hipsparse is needed @4.4.0:+rocm
hipsparse = spec["hipsparse"] hipsparse = spec["hipsparse"]
options += ["HIP_OPT=%s" % hipsparse.headers.cpp_flags] hip_headers += hipsparse.headers
hip_libs += hipsparse.libs hip_libs += hipsparse.libs
# Note: MFEM's defaults.mk wants to find librocsparse.* in # Note: MFEM's defaults.mk wants to find librocsparse.* in
# $(HIP_DIR)/lib, so we set HIP_DIR to be $ROCM_PATH when using # $(HIP_DIR)/lib, so we set HIP_DIR to be $ROCM_PATH when using
@ -943,11 +944,16 @@ def find_optional_library(name, prefix):
options += ["HIP_DIR=%s" % env["ROCM_PATH"]] options += ["HIP_DIR=%s" % env["ROCM_PATH"]]
else: else:
options += ["HIP_DIR=%s" % hipsparse["rocsparse"].prefix] options += ["HIP_DIR=%s" % hipsparse["rocsparse"].prefix]
if "^rocthrust" in spec and not spec["hip"].external:
# petsc+rocm needs the rocthrust header path
hip_headers += spec["rocthrust"].headers
if "%cce" in spec: if "%cce" in spec:
# We assume the proper Cray CCE module (cce) is loaded: # We assume the proper Cray CCE module (cce) is loaded:
craylibs_path = env["CRAYLIBS_" + machine().upper()] craylibs_path = env["CRAYLIBS_" + machine().upper()]
craylibs = ["libmodules", "libfi", "libcraymath", "libf", "libu", "libcsup"] craylibs = ["libmodules", "libfi", "libcraymath", "libf", "libu", "libcsup"]
hip_libs += find_libraries(craylibs, craylibs_path) hip_libs += find_libraries(craylibs, craylibs_path)
if hip_headers:
options += ["HIP_OPT=%s" % hip_headers.cpp_flags]
if hip_libs: if hip_libs:
options += ["HIP_LIB=%s" % ld_flags_from_library_list(hip_libs)] options += ["HIP_LIB=%s" % ld_flags_from_library_list(hip_libs)]