summaryrefslogtreecommitdiff
path: root/var/spack/repos/builtin/packages/arborx/package.py
diff options
context:
space:
mode:
Diffstat (limited to 'var/spack/repos/builtin/packages/arborx/package.py')
-rw-r--r--var/spack/repos/builtin/packages/arborx/package.py4
1 files changed, 4 insertions, 0 deletions
diff --git a/var/spack/repos/builtin/packages/arborx/package.py b/var/spack/repos/builtin/packages/arborx/package.py
index f461679c79..e617b59ee4 100644
--- a/var/spack/repos/builtin/packages/arborx/package.py
+++ b/var/spack/repos/builtin/packages/arborx/package.py
@@ -42,6 +42,7 @@ class Arborx(CMakePackage):
depends_on('cmake@3.12:', type='build')
depends_on('cmake@3.16:', type='build', when='@1.0:')
depends_on('mpi', when='+mpi')
+ depends_on('rocthrust', when='+rocm')
# Standalone Kokkos
depends_on('kokkos@3.1.00:', when='~trilinos')
@@ -72,6 +73,9 @@ class Arborx(CMakePackage):
# Only Kokkos allows '+cuda' for now
options.append(
'-DCMAKE_CXX_COMPILER=%s' % spec["kokkos"].kokkos_cxx)
+ if '+rocm' in spec:
+ options.append(
+ '-DCMAKE_CXX_COMPILER=%s' % spec["hip"].hipcc)
return options