From 6fbffcb6d64a44d72936b450cf8565c76f0f9991 Mon Sep 17 00:00:00 2001 From: Jian Sun Date: Fri, 4 Oct 2024 01:23:15 -0600 Subject: [PATCH] add new xml vars for GPU configuration --- cime_config/config_component.xml | 59 ++++++++++++++++++++++---------- 1 file changed, 40 insertions(+), 19 deletions(-) diff --git a/cime_config/config_component.xml b/cime_config/config_component.xml index b4d690a4..5d831eac 100644 --- a/cime_config/config_component.xml +++ b/cime_config/config_component.xml @@ -795,20 +795,38 @@ char - - + none,a100 + none build_def env_build.xml If set will compile and submit with this gpu type enabled - - char - - + + logical + TRUE,FALSE + FALSE build_def env_build.xml - If set will compile and submit with this gpu offload method enabled + True=>compile the GPU code with OpenACC GPU flags + + + + logical + TRUE,FALSE + FALSE + build_def + env_build.xml + True=>compile the GPU code with OpenMP GPU flags + + + + logical + TRUE,FALSE + FALSE + build_def + env_build.xml + True=>compile the GPU code with KOKKOS GPU target @@ -818,7 +836,7 @@ build_def env_build.xml If set will attach this script to the MPI run command, mapping - different MPI ranks to different GPUs within the same compute node + different MPI ranks to different GPUs within the same compute node @@ -1774,33 +1792,36 @@ integer 0 - - 1 - mach_pes_last env_mach_pes.xml - Number of CPU cores per GPU node used for simulation + Number of CPU cores per GPU node used for simulation + + + + logical + TRUE,FALSE + FALSE + mach_pes + env_mach_pes.xml + False=>assign only one MPI task per GPU; True=>assign multiple MPI tasks per GPU integer 0 - - 1 - mach_pes env_mach_pes.xml - Number of GPUs per node used for simulation + Number of GPUs per node used for simulation - + integer 0 mach_pes_last env_mach_pes.xml - maximum number of GPUs allowed per node + Maximum number of GPUs allowed per node - + integer $MAX_MPITASKS_PER_NODE