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

superlu-dist: fix rocm variant for the master branch (#33624)

Browse Source 
* superlu-dist: fix rocm variant for the master branch
* simplify superlu-dist
* add mpi include to HIP_HIPCC_FLAGS

Co-authored-by: liuyangzhuan <liuyangzhuan@users.noreply.github.com>
This commit is contained in:
liuyangzhuan 2022-11-01 10:28:01 -07:00 committed by GitHub
parent d925ba9bc6
commit 973b43b1c1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
var/spack/repos/builtin/packages/superlu-dist/package.py
View File

@ -122,12 +122,15 @@ def append_from_variant(*args):
if cuda_arch[0] != "none":
append_define("CMAKE_CUDA_ARCHITECTURES", cuda_arch[0])
if "+rocm" in spec and spec.satisfies("@amd"):
if "+rocm" in spec and (spec.satisfies("@amd") or spec.satisfies("@8:")):
append_define("TPL_ENABLE_HIPLIB", True)
append_define("HIP_ROOT_DIR", spec["hip"].prefix)
rocm_archs = spec.variants["amdgpu_target"].value
mpiinc = spec["mpi"].prefix.include
if "none" not in rocm_archs:
append_define("HIP_HIPCC_FLAGS", "--amdgpu-target=" + ",".join(rocm_archs))
append_define(
"HIP_HIPCC_FLAGS", "--amdgpu-target=" + ",".join(rocm_archs) + " -I/" + mpiinc
)
append_from_variant("BUILD_SHARED_LIBS", "shared")
return cmake_args