Forráskód Böngészése

savana_mem for slurm max mem

Thomas 6 napja
szülő
commit
22d9b9f29d
3 módosított fájl, 8 hozzáadás és 2 törlés
  1. 3 0
      pandora-config.example.toml
  2. 2 2
      src/callers/savana.rs
  3. 3 0
      src/config.rs

+ 3 - 0
pandora-config.example.toml

@@ -272,6 +272,9 @@ savana_bin = "/home/t_steimle/.conda/envs/savana_env/bin/savana"
 # Threads for Savana.
 savana_threads = 40
 
+# RAM capacity used for running Savana with slurm (in GB).
+savana_mem = 90
+
 # Savana output directory template.
 # {result_dir}, {id}
 savana_output_dir = "{result_dir}/{id}/diag/savana"

+ 2 - 2
src/callers/savana.rs

@@ -208,7 +208,7 @@ impl SlurmRunner for Savana {
         SlurmParams {
             job_name: Some(format!("savana_{}", self.id)),
             cpus_per_task: Some(self.config.savana_threads as u32),
-            mem: Some("80G".into()),
+            mem: Some(format!("{}G", self.config.savana_mem)),
             partition: Some("mediumq".into()),
             gres: None,
         }
@@ -221,7 +221,7 @@ impl SbatchRunner for Savana {
         SlurmParams {
             job_name: Some(format!("savana_{}", self.id)),
             cpus_per_task: Some(self.config.savana_threads as u32),
-            mem: Some("80G".into()),
+            mem: Some(format!("{}G", self.config.savana_mem)),
             partition: Some("mediumq".into()),
             gres: None,
         }

+ 3 - 0
src/config.rs

@@ -257,6 +257,9 @@ pub struct Config {
     /// Number of threads for Savana.
     pub savana_threads: u8,
 
+    /// RAM capacity used for running Savana with slurm.
+    pub savana_mem: u8,
+
     /// Template for Savana output directory (`{result_dir}`, `{id}`).
     pub savana_output_dir: String,