fv3gfs_remap_weights.sh

#! /usr/bin/env bash
#BSUB -L /bin/sh
#BSUB -P FV3GFS-T2O
#BSUB -oo log.weights
#BSUB -eo log.weights
#BSUB -J weights_fv3
#BSUB -q dev
#BSUB -M 1024
#BSUB -W 10:00
#BSUB -extsched 'CRAYLINUX[]'
set -ax

 . $MODULESHOME/init/sh
module load PrgEnv-intel
#--------------------------------------------------

export home_dir=/gpfs/hps3/emc/global/noscrub/$LOGNAME/git/fv3gfs/gfs.v15.0.0
export script_dir=$home_dir/ush
export exec_dir=$home_dir/exec
export fix_fv3_dir=$home_dir/fix
export fregrid=$home_dir/exec/fregrid_parallel
export TMPDIR=/gpfs/hps/ptmp/$LOGNAME/fv3_weight


#--global lat-lon array size
#----------------------------------------------------------
for GG in 1deg 0p5deg 0p25deg 0p125deg; do
#----------------------------------------------------------

if [ $GG = 1deg    ];  then  export nlon=360 ;  export nlat=180 ;fi  
if [ $GG = 0p5deg  ];  then  export nlon=720 ;  export nlat=360 ;fi  
if [ $GG = 0p25deg ];  then  export nlon=1440 ; export nlat=720 ;fi  
if [ $GG = 0p125deg ]; then  export nlon=2880 ; export nlat=1440 ;fi  

#----------------------------------------------------------
for CASE in C48  C96  C192  C384  C768  C1152  C3072; do
#----------------------------------------------------------
max_core=24
export NODES=3; export thread=1
##if [ $CASE = C3072 ]; then exportNODES=21; export thread=4; fi
export npes=$(((NODES-1)*max_core/thread))
 
export workdir=$TMPDIR/${CASE}_${GG}
mkdir -p $workdir; cd $workdir ||exit 8

export native_grid=$fix_fv3_dir/$CASE/${CASE}_mosaic.nc
export  remap_file=$fix_fv3_dir/$CASE/remap_weights_${CASE}_${GG}.nc

#NOTE: we are placing the first process on a node by itself to get the memory it needs
#      these npes will be tightly packed on the remaining nodes

aprun -n 1 -d 24 $fregrid --input_mosaic $native_grid \
                    --nlon $nlon --nlat $nlat \
                    --remap_file $remap_file \
                    --debug : \
   -n $npes -d $thread $fregrid --input_mosaic $native_grid \
                    --nlon $nlon --nlat $nlat \
                    --remap_file $remap_file \
                    --debug

done
done
exit