diff options
Diffstat (limited to 'var')
| -rw-r--r-- | var/spack/repos/builtin/packages/arborx/package.py | 13 |
1 files changed, 10 insertions, 3 deletions
diff --git a/var/spack/repos/builtin/packages/arborx/package.py b/var/spack/repos/builtin/packages/arborx/package.py index a4e559af86..e30279c3f2 100644 --- a/var/spack/repos/builtin/packages/arborx/package.py +++ b/var/spack/repos/builtin/packages/arborx/package.py @@ -6,7 +6,7 @@ from spack import * -class Arborx(CMakePackage): +class Arborx(CMakePackage, CudaPackage, ROCmPackage): """ArborX is a performance-portable library for geometric search""" homepage = "https://github.com/arborx/arborx" @@ -28,9 +28,7 @@ class Arborx(CMakePackage): # does not provide them. kokkos_backends = { 'serial': (True, "enable Serial backend (default)"), - 'cuda': (False, "enable Cuda backend"), 'openmp': (False, "enable OpenMP backend"), - 'rocm': (False, "enable HIP backend"), 'sycl': (False, "enable SYCL backend") } @@ -51,6 +49,15 @@ class Arborx(CMakePackage): for backend in kokkos_backends: depends_on('kokkos+%s' % backend.lower(), when='~trilinos+%s' % backend.lower()) + + for arch in CudaPackage.cuda_arch_values: + cuda_dep = "+cuda cuda_arch={0}".format(arch) + depends_on("kokkos {0}".format(cuda_dep), when=cuda_dep) + + for arch in ROCmPackage.amdgpu_targets: + rocm_dep = "+rocm amdgpu_target={0}".format(arch) + depends_on("kokkos {0}".format(rocm_dep), when=rocm_dep) + depends_on('kokkos+cuda_lambda', when='~trilinos+cuda') # Trilinos/Kokkos |