wcoss_preamble

#BSUB -W 2:30                    # wall clock time 
#BSUB -o C384C192_test_iau.stdout
#BSUB -e C384C192_test_iau.stderr
#BSUB -J C384C192_test_iau  
#BSUB -q "dev"                   # job queue 
#BSUB -P GFS-T2O                 # project code 
#BSUB -M 600                    # Memory req's for serial portion
#BSUB -extsched 'CRAYLINUX[]'    # Request to run on compute nodes
export NODES=40
export corespernode=24
export machine='wcoss'

# for control forecast
if [ $NODES -eq 20 ]; then
  # 20 nodes, 2 threads
  #export control_threads=2 # control forecast threads
  #export control_proc=444   # total number of processors for control forecast
  export control_threads=2
  export control_proc=444
  export write_groups_ctl=1 # write groups for control forecast.
  export layout_ctl="6,6" # layout_x,layout_y (total # mpi tasks = $layout_x*$layout_y*6=($fg_proc/$fg_threads) - $write_tasks*$write_groups)
elif [ $NODES -eq 40 ]; then
  # 40 nodes, 2 threads
  export control_threads=4 
  export control_proc=888  
  export write_groups_ctl=1
  export layout_ctl="6, 6"
elif [ $NODES -eq 80 ]; then
  # 80 nodes, 2 threads
  export control_threads=8
  export control_proc=1776 
  export write_groups_ctl=1
  export layout_ctl="12, 12" 
else
  echo "processor layout for $NODES nodes not set"
  exit 1
fi

# for ensemble forecast and GSI
export enkf_threads=2
export gsi_control_threads=2
export fg_proc=$corespernode 
export fg_threads=1 
export write_groups=1
export write_tasks=6 
export layout="3, 1"