From 109ed1d57b6b9d52bacc598ae9c0f72a4e46bc9e Mon Sep 17 00:00:00 2001
From: Matthias Schnepf <matthias.schnepf@kit.edu>
Date: Thu, 20 Mar 2025 14:18:51 +0100
Subject: [PATCH] Set memory and CPU via environment variables

---
 0212.node_worker_bwforcluster_nemo2.cfg | 22 +++++++++++++++-------
 1 file changed, 15 insertions(+), 7 deletions(-)

diff --git a/0212.node_worker_bwforcluster_nemo2.cfg b/0212.node_worker_bwforcluster_nemo2.cfg
index 8a5ac47..f968df0 100644
--- a/0212.node_worker_bwforcluster_nemo2.cfg
+++ b/0212.node_worker_bwforcluster_nemo2.cfg
@@ -10,8 +10,8 @@ STARTD.CloudSite = "BwForCluster-Nemo2"
 
 # Machine/Job environment
 # Use TardisDroneUuid to identify drones in OBS
-# TardisDroneUuid = "$ENV(TardisDroneUuid)"
-# STARTD_NAME = $ENV(TardisDroneUuid)
+TardisDroneUuid = "$ENV(TardisDroneUuid)"
+STARTD_NAME = "$ENV(TardisDroneUuid)"
 
 # Matchmaking advertisement
 STARTD.ProvidesIO    = TRUE
@@ -65,11 +65,19 @@ MASTER.DAEMON_SHUTDOWN = ( TimeToLive <= 10 * $(MINUTE) )
 MASTER.DAEMON_SHUTDOWN_FAST = ( TimeToLive <= 3 * $(MINUTE) )
 
 # Set number of CPUs
-NUM_CPUS = 32
-MEMORY = 128000
-
-TardisDroneUuid = "$ENV(TardisDroneUuid)"
-STARTD_NAME = $ENV(TardisDroneUuid)
+CPUsDefault = 8
+NUM_CPUS =  \
+    ifThenElse( isUndefined($ENV(SLURM_CPUS_ON_NODE)), \
+        $(CPUsDefault), \
+        $ENV(SLURM_CPUS_ON_NODE) \
+    )
+# Set memory
+MemoryDefault = 20480
+MEMORY = \
+    ifThenElse( isUndefined($ENV(SLURM_MEM_PER_NODE)), \
+        $(CPUsDefault), \
+        $ENV(SLURM_MEM_PER_NODE) \
+    )
 
 # The maximum number of claims during one negotiation cycle.
 # This is a tradeoff between allowing the cluster to fill fast (high value),
-- 
GitLab