From c2afb4b916940b5feeb4dbdb6eb744e21c84c9da Mon Sep 17 00:00:00 2001 From: eugeneswalker <38933153+eugeneswalker@users.noreply.github.com> Date: Fri, 6 May 2022 07:18:06 -0700 Subject: arborx: inherit from CudaPackage, ROCmPackage (#30490) --- var/spack/repos/builtin/packages/arborx/package.py | 13 ++++++++++--- 1 file 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 -- cgit v1.2.3-70-g09d2