Add back in kokkos options

E3SM-Project · Oct 9, 2023 · 2f2639c · 2f2639c
1 parent b39578d
commit 2f2639c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/cime_config/machines/cmake_macros/gnugpu_pm-gpu.cmake b/cime_config/machines/cmake_macros/gnugpu_pm-gpu.cmake
@@ -7,6 +7,7 @@ endif()
 string(APPEND CPPDEFS " -DTHRUST_IGNORE_CUB_VERSION_CHECK")
 string(APPEND SLIBS " -lblas -llapack")
 string(APPEND CUDA_FLAGS " -ccbin CC -O2 -arch sm_80 --use_fast_math")
+string(APPEND KOKKOS_OPTIONS " -DKokkos_ARCH_AMPERE80=On -DKokkos_ENABLE_CUDA=On -DKokkos_ENABLE_CUDA_LAMBDA=On -DKokkos_ENABLE_SERIAL=ON -DKokkos_ENABLE_OPENMP=Off")
 if (NOT DEBUG)
   string(APPEND CFLAGS " -O2")
   string(APPEND FFLAGS " -O2")