Skip to content

Commit

Permalink
ATDM: Make KOKKOS_ARCH looping work with commas (trilinos#7247)
Browse files Browse the repository at this point in the history
  • Loading branch information
bartlettroscoe committed May 28, 2020
1 parent 87da239 commit 8ea5106
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions cmake/std/atdm/ATDMDevEnvSettings.cmake
Original file line number Diff line number Diff line change
Expand Up @@ -294,9 +294,10 @@ ATDM_SET_CACHE(Kokkos_ENABLE_CUDA_LAMBDA "${ATDM_USE_CUDA}" CACHE BOOL)
ATDM_SET_CACHE(Kokkos_ENABLE_DEBUG_BOUNDS_CHECK "${Trilinos_ENABLE_DEBUG}" CACHE BOOL)
ATDM_SET_CACHE(Kokkos_ENABLE_DEBUG "${Trilinos_ENABLE_DEBUG}" CACHE BOOL)

# Update Kokkos arch, expected ATDM_CONFIG_KOKKOS_ARCH=arch1;arch2;arch3
# set the modern Kokkos arch, e.g., KOKKOS_ARCH=arch1 => Kokkos_ARCH_arch1=ON
# Update Kokkos arch, expected ATDM_CONFIG_KOKKOS_ARCH=arch1,arch2,arch3. Set
# the modern Kokkos arch, e.g., KOKKOS_ARCH=arch1 => Kokkos_ARCH_arch1=ON
set(kokkos_arch_list "$ENV{ATDM_CONFIG_KOKKOS_ARCH}")
string(REPLACE "," ";" kokkos_arch_list "$ENV{ATDM_CONFIG_KOKKOS_ARCH}")
foreach(kokkos_arch_loop_var ${kokkos_arch_list} )
ATDM_SET_ENABLE(Kokkos_ARCH_${kokkos_arch_loop_var} ON)
endforeach(kokkos_arch_loop_var)
Expand Down

0 comments on commit 8ea5106

Please sign in to comment.