From 5699e3c8cf43e6839c636eabfbb60ba148b341ac Mon Sep 17 00:00:00 2001
From: Axel Kohlmeyer <akohlmey@gmail.com>
Date: Thu, 12 Sep 2024 17:34:12 -0400
Subject: [PATCH] add overridable settings

---
 cmake/Modules/Packages/KOKKOS.cmake | 17 +++++++++++++++--
 1 file changed, 15 insertions(+), 2 deletions(-)

diff --git a/cmake/Modules/Packages/KOKKOS.cmake b/cmake/Modules/Packages/KOKKOS.cmake
index 8c5b1229b5..08b109eff4 100644
--- a/cmake/Modules/Packages/KOKKOS.cmake
+++ b/cmake/Modules/Packages/KOKKOS.cmake
@@ -10,8 +10,21 @@ endif()
 if(Kokkos_ENABLE_CUDA)
   message(STATUS "KOKKOS: Enabling CUDA LAMBDA function support")
   set(Kokkos_ENABLE_CUDA_LAMBDA ON CACHE BOOL "" FORCE)
-  message(STATUS "KOKKOS: Disabling CUDA malloc async support")
-  set(Kokkos_ENABLE_IMPL_CUDA_MALLOC_ASYNC OFF CACHE BOOL "" FORCE)
+  option(Kokkos_ENABLE_IMPL_CUDA_MALLOC_ASYNC "CUDA asynchronous malloc support" OFF)
+  mark_as_advanced(Kokkos_ENABLE_IMPL_CUDA_MALLOC_ASYNC)
+  if(Kokkos_ENABLE_IMPL_CUDA_MALLOC_ASYNC)
+    message(STATUS "KOKKOS: CUDA malloc async support enabled")
+  else()
+    message(STATUS "KOKKOS: CUDA malloc async support disabled")
+  endif()
+endif()
+if(Kokkos_ENABLE_HIP)
+  option(KOKKOS_ENABLE_IMPL_HIP_UNIFIED_MEMORY "Enable unified memory with HIP" ON)
+  mark_as_advanced(KOKKOS_ENABLE_IMPL_HIP_UNIFIED_MEMORY)
+  option(KOKKOS_ENABLE_HIP_MULTIPLE_KERNEL_INSTANTIATIONS "Enable multiple kernel instantiations with HIP" ON)
+  mark_as_advanced(KOKKOS_ENABLE_HIP_MULTIPLE_KERNEL_INSTANTIATIONS)
+  option(KOKKOS_ENABLE_ROCTHRUST "Use RoCThrust library" ON)
+  mark_as_advanced(KOKKOS_ENABLE_ROCTHRUST)
 endif()
 # Adding OpenMP compiler flags without the checks done for
 # BUILD_OMP can result in compile failures. Enforce consistency.