Changeset 7685 for branches/UKMO/dev_r7681_rebuild_nemo/NEMOGCM/TOOLS/REBUILD_NEMO/rebuild_nemo

Timestamp:

2017-02-15T17:24:10+01:00 (7 years ago)

Author:

timgraham

Message:

Added option to specify required memory for jobs submitted to a compute node

File:

• branches/UKMO/dev_r7681_rebuild_nemo/NEMOGCM/TOOLS/REBUILD_NEMO/rebuild_nemo (modified) (5 diffs)

branches/UKMO/dev_r7681_rebuild_nemo/NEMOGCM/TOOLS/REBUILD_NEMO/rebuild_nemo

@@ -18 +18 @@
    echo "            -d deflate_level   deflate level for output files (key_netcdf4 only)"
    echo "            -n namelist        full path to namelist file to be created (otherwise default nam_rebuild+_process_id is used)"
+   echo "            -r memory          Memory to request on compute node including units (Default = 10Gb)"
    echo
    exit 1
 }
 
-while getopts c:n:t:d:lm opt
+while getopts c:n:t:d:r:lm opt
 do
   case ${opt} in
@@ -39 +40 @@
       m) 
          NMASK="TRUE"
+         echo "mask"
       ;;
       d)
@@ -45 +47 @@
       n)
          nam_rebuild=${OPTARG}
+      ;;
+      r)
+         MEMORY=${OPTARG}
       ;;
   esac
@@ -62 +67 @@
 export OMP_NUM_THREADS=${OMP_NUM_THREADS:-1}
 nam_rebuild=${nam_rebuild:-nam_rebuild_$$}
+MEMORY=${MEMORY:-10Gb}
 
 if [[ -n ${DIM1} && -n ${DIM2} ]] ; then
@@ -96 +102 @@
    #The process ID is appended to the end of the file name so it is unique
    cat ${script_dir}/BATCH_TEMPLATES/${batch_file} | sed -e"s/NTHREADS/${OMP_NUM_THREADS}/" \
+      -e"s/MEMORY/${MEMORY}/" \
       -e"s:INDIR:${script_dir}:" \
       -e"s/NAMELIST/${nam_rebuild}/" > ${batch_file}_$$.sh