From 79e6002bae54e6a956b6acdd7ac9827bb8657085 Mon Sep 17 00:00:00 2001
From: Jim Edwards <jedwards@ucar.edu>
Date: Wed, 9 Oct 2024 14:20:54 -0600
Subject: [PATCH] add memory usage variables for use on derecho

---
 machines/derecho/config_batch.xml    | 2 +-
 machines/derecho/config_machines.xml | 2 ++
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/machines/derecho/config_batch.xml b/machines/derecho/config_batch.xml
index cb83e8e2..406411be 100644
--- a/machines/derecho/config_batch.xml
+++ b/machines/derecho/config_batch.xml
@@ -5,7 +5,7 @@
   </submit_args>
   <directives gpu_enabled="false">
     <directive default="/bin/bash" > -S {{ shell }}  </directive>
-    <directive> -l select={{ num_nodes }}:ncpus={{ max_tasks_per_node }}:mpiprocs={{ tasks_per_node }}:ompthreads={{ thread_count }}</directive>
+    <directive> -l select={{ num_nodes }}:ncpus={{ max_tasks_per_node }}:mpiprocs={{ tasks_per_node }}:ompthreads={{ thread_count }}:mem={{ mem_per_node }}GB</directive>
   </directives>
   <directives gpu_enabled="true">
     <directive default="/bin/bash" > -S {{ shell }} </directive>
diff --git a/machines/derecho/config_machines.xml b/machines/derecho/config_machines.xml
index 12446f37..d53c181f 100644
--- a/machines/derecho/config_machines.xml
+++ b/machines/derecho/config_machines.xml
@@ -13,6 +13,8 @@
     <BATCH_SYSTEM>pbs</BATCH_SYSTEM>
     <SUPPORTED_BY>cseg</SUPPORTED_BY>
     <MAX_TASKS_PER_NODE>128</MAX_TASKS_PER_NODE>
+    <MEM_PER_TASK>10</MEM_PER_TASK>
+    <MAX_MEM_PER_NODE>235</MAX_MEM_PER_NODE>
     <MAX_GPUS_PER_NODE>4</MAX_GPUS_PER_NODE>
     <MAX_MPITASKS_PER_NODE>128</MAX_MPITASKS_PER_NODE>
     <MAX_CPUTASKS_PER_GPU_NODE>64</MAX_CPUTASKS_PER_GPU_NODE>