diff --git a/var/spack/repos/builtin/packages/hwloc/package.py b/var/spack/repos/builtin/packages/hwloc/package.py index 995f027e5b8..0f412b4a122 100644 --- a/var/spack/repos/builtin/packages/hwloc/package.py +++ b/var/spack/repos/builtin/packages/hwloc/package.py @@ -156,6 +156,11 @@ def configure_args(self): if '+rocm' not in self.spec: args.append('--disable-rsmi') + if '+rocm' in self.spec: + args.append('--with-rocm={0}'.format(self.spec['rocm'].prefix)) + args.append('--with-rocm-version={0}'.format( + self.spec['rocm'].version)) + if '+netloc' in self.spec: args.append('--enable-netloc') @@ -168,4 +173,9 @@ def configure_args(self): args.extend(self.enable_or_disable('pci')) args.extend(self.enable_or_disable('shared')) + if '+cuda' in self.spec: + args.append('--with-cuda={0}'.format(self.spec['cuda'].prefix)) + args.append('--with-cuda-version={0}'.format( + self.spec['cuda'].version)) + return args