Add PBS environment detection - PBS_RESC_MEM and PBS_NCPUS. Issue #1912.

marbl · Mar 31, 2021 · 404540a · 404540a
1 parent 8ebd75b
commit 404540a
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/src/pipelines/canu/Grid_Local.pm b/src/pipelines/canu/Grid_Local.pm
@@ -68,6 +68,10 @@ sub getNumberOfCPUs () {
         $ncpu = $ENV{"SLURM_JOB_CPUS_PER_NODE"};
     }
 
+    if (exists($ENV{"PBS_NCPUS"})) {
+        $ncpu = $ENV{"PBS_NCPUS"};
+    }
+
     if (exists($ENV{"PBS_NUM_PPN"})) {
         $ncpu = $ENV{"PBS_NUM_PPN"};
     }
@@ -114,6 +118,10 @@ sub getPhysicalMemorySize () {
         $memory = $ENV{"SLURM_MEM_PER_NODE"};
     }
 
+    if (exists($ENV{"PBS_RESC_MEM"})) {
+        $memory = $ENV{"PBS_RESC_MEM"} / 1024 / 1024 / 1024;
+    }
+
     return(int($memory + 0.5));  #  Poor man's rounding
 }
 

diff --git a/src/utility b/src/utility