From cbc867a24c51b7137768c1c20685c8c9e6cef1fa Mon Sep 17 00:00:00 2001 From: Sarah Osborn <30503782+osborn9@users.noreply.github.com> Date: Mon, 10 Oct 2022 06:46:26 -0700 Subject: hypre: fix to correctly find rocsparse and rocrand when not in ROCM_PATH (#33073) --- var/spack/repos/builtin/packages/hypre/package.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) (limited to 'var') diff --git a/var/spack/repos/builtin/packages/hypre/package.py b/var/spack/repos/builtin/packages/hypre/package.py index 36d4ec39df..874b2321cb 100644 --- a/var/spack/repos/builtin/packages/hypre/package.py +++ b/var/spack/repos/builtin/packages/hypre/package.py @@ -93,7 +93,9 @@ class Hypre(AutotoolsPackage, CudaPackage, ROCmPackage): depends_on("blas") depends_on("lapack") depends_on("superlu-dist", when="+superlu-dist+mpi") - + depends_on("rocsparse", when="+rocm") + depends_on("rocthrust", when="+rocm") + depends_on("rocrand", when="+rocm") depends_on("umpire", when="+umpire") for sm_ in CudaPackage.cuda_arch_values: depends_on( @@ -235,11 +237,16 @@ class Hypre(AutotoolsPackage, CudaPackage, ROCmPackage): configure_args.append("--disable-cub") if "+rocm" in spec: + rocm_pkgs = ["rocsparse", "rocthrust", "rocprim", "rocrand"] + rocm_inc = "" + for pkg in rocm_pkgs: + rocm_inc += spec[pkg].headers.include_flags + " " configure_args.extend( [ "--with-hip", "--enable-rocrand", "--enable-rocsparse", + "--with-extra-CUFLAGS={0}".format(rocm_inc), ] ) rocm_arch_vals = spec.variants["amdgpu_target"].value -- cgit v1.2.3-70-g09d2