ATDM: ride: Spread out work over GPUs (trilinos#2422)

This also switches to patched CMake 3.17.2 which is needed to support this feature.
KyleFromKitware · May 20, 2020 · 88cd592 · 88cd592
1 parent 25d209a
commit 88cd592
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/cmake/std/atdm/ride/environment.sh b/cmake/std/atdm/ride/environment.sh
@@ -225,7 +225,7 @@ export ATDM_CONFIG_NETCDF_LIBS="-L${NETCDF_ROOT}/lib;-L${HDF5_ROOT}/lib;${NETCDF
 
 # Use manually installed cmake and ninja to try to avoid module loading
 # problems (see TRIL-208)
-export PATH=/ascldap/users/rabartl/install/white-ride/cmake-3.11.2/bin:/ascldap/users/rabartl/install/white-ride/ninja-1.8.2/bin:$PATH
+export PATH=/home/atdm-devops-admin/tools/ride/cmake-3.17.2/bin:/home/rabartl/install/white-ride/ninja-1.8.2/bin:$PATH
 
 # Set MPI wrappers
 export MPICC=`which mpicc`

diff --git a/cmake/std/atdm/ride/tweaks/Tweaks.cmake b/cmake/std/atdm/ride/tweaks/Tweaks.cmake
@@ -1,3 +1,11 @@
+#
+# Set up to limit running on GPUs
+#
+
+ATDM_SET_CACHE(Trilinos_AUTOGENERATE_TEST_RESOURCE_FILE ON CACHE BOOL)
+ATDM_SET_CACHE(Trilinos_CUDA_NUM_GPUS 2 CACHE STRING)
+ATDM_SET_CACHE(Trilinos_CUDA_SLOTS_PER_GPU 2 CACHE STRING)
+
 #
 # Disables across multiple builds on 'ride'
 #