diff --git a/var/spack/repos/builtin/packages/superlu-dist/package.py b/var/spack/repos/builtin/packages/superlu-dist/package.py index 21a5da15421..d429da8ce92 100644 --- a/var/spack/repos/builtin/packages/superlu-dist/package.py +++ b/var/spack/repos/builtin/packages/superlu-dist/package.py @@ -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