Merge pull request #174 from xylar/add-gpus-resource

Add `gpus_per_node` for machines with GPUs
E3SM-Project · Jul 12, 2024 · ea9098f · ea9098f
2 parents bae6e7e + b3bce82
commit ea9098f
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/mache/machines/frontier.cfg b/mache/machines/frontier.cfg
@@ -40,9 +40,12 @@ system = slurm
 # whether to use mpirun or srun to run a task
 parallel_executable = srun
 
-# cores per node on the machine
+# allocatable cores per node on the machine
 cores_per_node = 56
 
+# gpus per node on the machine
+gpus_per_node = 4
+
 # account for running diagnostics jobs
 account = cli115
 

diff --git a/mache/machines/pm-gpu.cfg b/mache/machines/pm-gpu.cfg
@@ -50,8 +50,11 @@ system = slurm
 # whether to use mpirun or srun to run a task
 parallel_executable = srun
 
-# cores per node on the machine
-cores_per_node = 256
+# cores per node on the machine (with hyperthreading)
+cores_per_node = 128
+
+# gpus per node on the machine
+gpus_per_node = 4
 
 # account for running diagnostics jobs
 account = e3sm