Skip to content

Commit

Permalink
Add initial config files for global aerosol DA (#986)
Browse files Browse the repository at this point in the history
Add new config files for aerosol da
  • Loading branch information
CoryMartin-NOAA authored Aug 19, 2022
1 parent cfcc21f commit 8a62c3a
Show file tree
Hide file tree
Showing 7 changed files with 115 additions and 0 deletions.
11 changes: 11 additions & 0 deletions env/HERA.env
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,7 @@ if [ $# -ne 1 ]; then
echo "Must specify an input argument to set runtime environment variables!"
echo "argument can be any one of the following:"
echo "atmanalrun atmensanalrun"
echo "aeroanlrun"
echo "anal sfcanl fcst post vrfy metp"
echo "eobs eupd ecen efcs epos"
echo "postsnd awips gempak"
Expand Down Expand Up @@ -75,6 +76,16 @@ elif [ $step = "atmensanalrun" ]; then
[[ $NTHREADS_ATMENSANAL -gt $nth_max ]] && export NTHREADS_ATMENSANAL=$nth_max
export APRUN_ATMENSANAL="$launcher -n $npe_atmensanalrun"

elif [ $step = "aeroanlrun" ]; then

export APRUNCFP="$launcher -n \$ncmd --multi-prog"

nth_max=$(($npe_node_max / $npe_node_aeroanlrun))

export NTHREADS_AEROANL=${nth_aeroanlrun:-$nth_max}
[[ $NTHREADS_AEROANL -gt $nth_max ]] && export NTHREADS_AEROANL=$nth_max
export APRUN_AEROANL="$launcher -n $npe_aeroanlrun"

elif [ $step = "anal" ]; then

export MKL_NUM_THREADS=4
Expand Down
11 changes: 11 additions & 0 deletions env/ORION.env
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,7 @@ if [ $# -ne 1 ]; then
echo "Must specify an input argument to set runtime environment variables!"
echo "argument can be any one of the following:"
echo "atmanalrun atmensanalrun"
echo "aeroanlrun"
echo "anal sfcanl fcst post vrfy metp"
echo "eobs eupd ecen efcs epos"
echo "postsnd awips gempak"
Expand Down Expand Up @@ -74,6 +75,16 @@ elif [ $step = "atmensanalrun" ]; then
[[ $NTHREADS_ATMENSANAL -gt $nth_max ]] && export NTHREADS_ATMENSANAL=$nth_max
export APRUN_ATMENSANAL="$launcher -n $npe_atmensanalrun"

elif [ $step = "aeroanlrun" ]; then

export APRUNCFP="$launcher -n \$ncmd --multi-prog"

nth_max=$(($npe_node_max / $npe_node_aeroanlrun))

export NTHREADS_AEROANL=${nth_aeroanlrun:-$nth_max}
[[ $NTHREADS_AEROANL -gt $nth_max ]] && export NTHREADS_AEROANL=$nth_max
export APRUN_AEROANL="$launcher -n $npe_aeroanlrun"

elif [ $step = "anal" ]; then

export MKL_NUM_THREADS=4
Expand Down
17 changes: 17 additions & 0 deletions parm/config/config.aeroanl
Original file line number Diff line number Diff line change
@@ -0,0 +1,17 @@
#!/bin/bash -x

########## config.aeroanl ##########
# configuration common to all aero analysis tasks

echo "BEGIN: config.aeroanl"

export OBS_YAML_DIR=$HOMEgfs/sorc/gdas.cd/parm/aero/obs/config/
export OBS_LIST=$HOMEgfs/sorc/gdas.cd/parm/aero/obs/lists/aero_prototype.yaml
export AEROVARYAML=$HOMEgfs/sorc/gdas.cd/parm/aero/variational/3dvar_dripcg.yaml
export BERROR_YAML=$HOMEgfs/sorc/gdas.cd/parm/aero/berror/static_bump.yaml
export FV3JEDI_FIX=$HOMEgfs/fix/fix_jedi

export io_layout_x=1
export io_layout_y=1

echo "END: config.aeroanl"
10 changes: 10 additions & 0 deletions parm/config/config.aeroanlfinal
Original file line number Diff line number Diff line change
@@ -0,0 +1,10 @@
#!/bin/bash -x

########## config.aeroanlfinal ##########
# Post Aero Analysis specific

echo "BEGIN: config.aeroanlfinal"

# Get task specific resources
. $EXPDIR/config.resources aeroanlfinal
echo "END: config.aeroanlfinal"
10 changes: 10 additions & 0 deletions parm/config/config.aeroanlinit
Original file line number Diff line number Diff line change
@@ -0,0 +1,10 @@
#!/bin/bash -x

########## config.aeroanlinit ##########
# Pre Aero Analysis specific

echo "BEGIN: config.aeroanlinit"

# Get task specific resources
. $EXPDIR/config.resources aeroanlinit
echo "END: config.aeroanlinit"
14 changes: 14 additions & 0 deletions parm/config/config.aeroanlrun
Original file line number Diff line number Diff line change
@@ -0,0 +1,14 @@
#!/bin/bash -x

########## config.aeroanlrun ##########
# Aerosol Analysis specific

echo "BEGIN: config.aeroanlrun"

# Get task specific resources
. $EXPDIR/config.resources aeroanlrun

# Task specific variables
export JEDIVAREXE=$HOMEgfs/exec/fv3jedi_var.x

echo "END: config.aeroanlrun"
42 changes: 42 additions & 0 deletions parm/config/config.resources
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,7 @@ if [ $# -ne 1 ]; then
echo "getic init coupled_ic aerosol_init"
echo "atmanalprep atmanalrun atmanalpost"
echo "atmensanalprep atmensanalrun atmensanalpost"
echo "aeroanlinit aeroanlrun aeroanlfinal"
echo "anal sfcanl analcalc analdiag gldas fcst post vrfy metp arch echgres"
echo "eobs ediag eomg eupd ecen esfc efcs epos earc"
echo "init_chem mom6ic ocnpost"
Expand Down Expand Up @@ -156,6 +157,47 @@ elif [ $step = "atmanalpost" ]; then
export nth_atmanalpost=1
export npe_node_atmanalpost=$(echo "$npe_node_max / $nth_atmanalpost" | bc)

elif [ $step = "aeroanlinit" ]; then

export wtime_aeroanlinit="00:10:00"
export npe_aeroanlinit=1
export nth_aeroanlinit=1
export npe_node_aeroanlinit=$(echo "$npe_node_max / $nth_aeroanlinit" | bc)
export memory_aeroanlinit="3072M"

elif [ $step = "aeroanlrun" ]; then

case $CASE in
C768)
layout_x=6
layout_y=6
;;
C384)
layout_x=5
layout_y=5
;;
C192 | C96 | C48)
layout_x=3
layout_y=3
;;
esac

export wtime_aeroanlrun="00:30:00"
export npe_aeroanlrun=$(echo "$layout_x * $layout_y * 6" | bc)
export npe_aeroanlrun_gfs=$(echo "$layout_x * $layout_y * 6" | bc)
export nth_aeroanlrun=1
export nth_aeroanlrun_gfs=1
export native_aeroanlrun="--exclusive"
export npe_node_aeroanlrun=$(echo "$npe_node_max / $nth_aeroanlrun" | bc)

elif [ $step = "aeroanlfinal" ]; then

export wtime_aeroanlfinal="00:10:00"
export npe_aeroanlfinal=1
export nth_aeroanlfinal=1
export npe_node_aeroanlfinal=$(echo "$npe_node_max / $nth_aeroanlfinal" | bc)
export memory_aeroanlfinal="3072M"

elif [ $step = "anal" ]; then

export wtime_anal="01:00:00"
Expand Down

0 comments on commit 8a62c3a

Please sign in to comment.