summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--var/spack/repos/builtin/packages/hwloc/package.py10
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