Build GPU code only for arch enabled in Kokkos

CMakeLists.txt

@@ -7,6 +7,20 @@ if(Kokkos_ENABLE_CUDA)
   kokkos_check(OPTIONS CUDA_LAMBDA)
 endif()
 
+if(Kokkos_ENABLE_HIP AND NOT DEFINED GPU_TARGETS)
+  if(VEGA900 IN_LIST Kokkos_ARCH)
+    set(GPU_TARGETS gfx900)
+  elseif(VEGA906 IN_LIST Kokkos_ARCH)
+    set(GPU_TARGETS gfx906)
+  elseif(VEGA908 IN_LIST Kokkos_ARCH)
+    set(GPU_TARGETS gfx908)
+  elseif(VEGA90A IN_LIST Kokkos_ARCH)
+    set(GPU_TARGETS gfx90a)
+  else()
+    message(WARNING "GPU device code will be built for all supported architectures for AMD ROCm.")
+  endif()
+endif()
+
 add_library(ArborX INTERFACE)
 target_link_libraries(ArborX INTERFACE Kokkos::kokkos)
 set_target_properties(ArborX PROPERTIES INTERFACE_COMPILE_FEATURES cxx_std_14)