Skip to content

Commit

Permalink
Merge pull request #363 from sjsprecious/add_gpu_gust
Browse files Browse the repository at this point in the history
Add new XML variables for flexible controls of GPU configuration
  • Loading branch information
jedwards4b authored Aug 9, 2023
2 parents 65770e1 + 72ee0b2 commit e951fdc
Showing 1 changed file with 40 additions and 2 deletions.
42 changes: 40 additions & 2 deletions cime_config/config_component.xml
Original file line number Diff line number Diff line change
Expand Up @@ -784,6 +784,34 @@
If TRUE, the component libraries are always built with OpenMP capability.</desc>
</entry>

<entry id="GPU_TYPE">
<type>char</type>
<valid_values></valid_values>
<default_value></default_value>
<group>build_def</group>
<file>env_build.xml</file>
<desc>If set will compile and submit with this gpu type enabled </desc>
</entry>

<entry id="GPU_OFFLOAD">
<type>char</type>
<valid_values></valid_values>
<default_value></default_value>
<group>build_def</group>
<file>env_build.xml</file>
<desc>If set will compile and submit with this gpu offload method enabled </desc>
</entry>

<entry id="MPI_GPU_WRAPPER_SCRIPT">
<type>char</type>
<valid_values></valid_values>
<default_value></default_value>
<group>build_def</group>
<file>env_build.xml</file>
<desc>If set will attach this script to the MPI run command, mapping
different MPI ranks to different GPUs within the same compute node</desc>
</entry>

<entry id="SMP_PRESENT">
<type>logical</type>
<valid_values>TRUE,FALSE</valid_values>
Expand Down Expand Up @@ -1798,12 +1826,22 @@
<desc>pes or cores per node for accounting purposes </desc>
</entry>

<entry id="MAX_CPUTASKS_PER_GPU_NODE">
<type>integer</type>
<default_value>0</default_value>
<values>
<value compiler="nvhpc">1</value>
</values>
<group>mach_pes_last</group>
<file>env_mach_pes.xml</file>
<desc> Number of CPU cores per GPU node used for simulation </desc>
</entry>

<entry id="NGPUS_PER_NODE">
<type>integer</type>
<default_value>0</default_value>
<values>
<value compiler="pgi-gpu">1</value>
<value compiler="nvhpc-gpu">1</value>
<value compiler="nvhpc">1</value>
</values>
<group>mach_pes</group>
<file>env_mach_pes.xml</file>
Expand Down

0 comments on commit e951fdc

Please sign in to comment.