diff options
| -rw-r--r-- | var/spack/repos/builtin/packages/hwloc/package.py | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/var/spack/repos/builtin/packages/hwloc/package.py b/var/spack/repos/builtin/packages/hwloc/package.py index 995f027e5b..0f412b4a12 100644 --- a/var/spack/repos/builtin/packages/hwloc/package.py +++ b/var/spack/repos/builtin/packages/hwloc/package.py @@ -156,6 +156,11 @@ class Hwloc(AutotoolsPackage): 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 @@ class Hwloc(AutotoolsPackage): 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 |