kokkos-ndevices flag is deprecated, replacing with kokkos-map-device-…

…id-by=mpi_rank
sandialabs · Oct 20, 2022 · 7bf9b54 · 7bf9b54
1 parent d8321dc
commit 7bf9b54
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -762,9 +762,7 @@ IF(ALBANY_ENABLE_CUDA)
   IF (ENABLE_ALBANY_EPETRA)
     MESSAGE(FATAL_ERROR "\nError: Albany with Epetra is not supported with CUDA KokkosNode!  Please re-configure w/o Epetra (ENABLE_ALBANY_EPETRA=OFF).")
   ENDIF()
-  SET(NUM_GPUS_PER_NODE 4 CACHE STRING "set Number of GPUs per Node")
-  MESSAGE("-- NUM_GPUS_PER_NODE=${NUM_GPUS_PER_NODE}")
-  SET(KOKKOS_NDEVICES "--kokkos-ndevices=${NUM_GPUS_PER_NODE}")
+  SET(KOKKOS_NDEVICES "--kokkos-map-device-id-by=mpi_rank")
   IF (ENABLE_CUDA_ERROR_CHECK)
     SET(ALBANY_CUDA_ERROR_CHECK ON)
     MESSAGE("-- CUDA Error Check        Enabled")