Skip to content

Commit

Permalink
use AMD-EPYC-based spack-stack
Browse files Browse the repository at this point in the history
  • Loading branch information
weihuang-jedi committed Dec 5, 2024
1 parent 7680a23 commit b6458c7
Show file tree
Hide file tree
Showing 13 changed files with 97 additions and 14 deletions.
2 changes: 1 addition & 1 deletion modulefiles/module_gwci.noaacloud.lua
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@ help([[
Load environment to run GFS workflow setup scripts on noaacloud
]])

prepend_path("MODULEPATH", "/contrib/spack-stack/spack-stack-1.6.0/envs/unified-env/install/modulefiles/Core")
prepend_path("MODULEPATH", "/contrib/Natalie.Perlin/spack-stack/spack-stack-1.6.0/envs/ue-intel/install/modulefiles/Core")

load(pathJoin("stack-intel", os.getenv("2021.3.0")))
load(pathJoin("stack-intel-oneapi-mpi", os.getenv("2021.3.0")))
Expand Down
4 changes: 2 additions & 2 deletions modulefiles/module_gwsetup.noaacloud.lua
Original file line number Diff line number Diff line change
Expand Up @@ -4,8 +4,8 @@ Load environment to run GFS workflow setup scripts on noaacloud

load(pathJoin("rocoto"))

prepend_path("MODULEPATH", "/contrib/spack-stack-rocky8/spack-stack-1.6.0/envs/ue-intel/install/modulefiles/Core")
prepend_path("MODULEPATH", "/apps/modules/modulefiles")
prepend_path("MODULEPATH", "/contrib/Natalie.Perlin/spack-stack/spack-stack-1.6.0/envs/ue-intel/install/modulefiles/Core")

load("gnu")
load("stack-intel")
load("stack-intel-oneapi-mpi")
Expand Down
2 changes: 1 addition & 1 deletion parm/config/gefs/config.resources
Original file line number Diff line number Diff line change
Expand Up @@ -41,7 +41,7 @@ case ${machine} in
;;
"AWSPW")
export PARTITION_BATCH="compute"
max_tasks_per_node=36
max_tasks_per_node=48
;;
"AZUREPW")
export PARTITION_BATCH="compute"
Expand Down
8 changes: 4 additions & 4 deletions scripts/exglobal_cleanup.sh
Original file line number Diff line number Diff line change
Expand Up @@ -8,14 +8,14 @@ echo "Begin Cleanup ${DATAROOT}!"
# Remove DATAoutput from the forecast model run
# TODO: Handle this better
DATAfcst="${DATAROOT}/${RUN}fcst.${PDY:-}${cyc}"
if [[ -d "${DATAfcst}" ]]; then rm -rf "${DATAfcst}"; fi
#if [[ -d "${DATAfcst}" ]]; then rm -rf "${DATAfcst}"; fi
#DATAefcs="${DATAROOT}/${RUN}efcs???${PDY:-}${cyc}"
rm -rf "${DATAROOT}/${RUN}efcs"*"${PDY:-}${cyc}"
#rm -rf "${DATAROOT}/${RUN}efcs"*"${PDY:-}${cyc}"
###############################################################

if [[ "${CLEANUP_COM:-YES}" == NO ]] ; then
#if [[ "${CLEANUP_COM:-YES}" == NO ]] ; then
exit 0
fi
#fi

###############################################################
# Clean up previous cycles; various depths
Expand Down
5 changes: 5 additions & 0 deletions scripts/exglobal_forecast.sh
Original file line number Diff line number Diff line change
Expand Up @@ -149,6 +149,11 @@ else
export OMP_NUM_THREADS=${UFS_THREADS:-1}
fi

#module load slurm

#export I_MPI_FABRICS=shm:ofi
#export I_MPI_OFI_PROVIDER=tcp

${NCP} "${EXECgfs}/${FCSTEXEC}" "${DATA}/"
${APRUN_UFS} "${DATA}/${FCSTEXEC}" 1>&1 2>&2
export ERR=$?
Expand Down
13 changes: 13 additions & 0 deletions sfsc96.sh
Original file line number Diff line number Diff line change
@@ -0,0 +1,13 @@
#!/bin/bash

set -x

source ~/.bashrc
source workflow/gw_setup.sh

HPC_ACCOUNT=${USER} \
pslot=c96sfs \
RUNTESTS=/lustre/$USER/run \
./workflow/create_experiment.py \
--yaml sfsc96/sfsc96.yaml

39 changes: 39 additions & 0 deletions sfsc96/options2sfsc96.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,39 @@
base:
DO_JEDIATMVAR: "NO"
DO_JEDIATMENS: "NO"
DO_JEDIOCNVAR: "NO"
DO_JEDISNOWDA: "NO"
DO_MERGENSST: "NO"
DO_BUFRSND: "NO"
DO_GEMPAK: "NO"
DO_AWIPS: "NO"
WRITE_DOPOST: "YES"
KEEPDATA: "YES"
DO_EXTRACTVARS: "NO"
FHMAX_GFS: 744
FHMAX_HF_GFS: 744
FHOUT_HF_GFS: 24
FHOUT_GFS: 24
FHOUT_OCN_GFS: 24
FHOUT_ICE_GFS: 24
FCST_BREAKPOINTS: ""
REPLAY_ICS: "NO"
RERUN: "NO"
HPSSARCH: "NO"
LOCALARCH: "YES"
ATSRDIR: "/bucket/Wei.Huang/SFS-REPLAY/archive/sfsc96/1994050100"
FLTFILEGFS: "postxconfig-NT-SFS.txt"
FLTFILEGFSF00: "postxconfig-NT-SFS.txt"
ACCOUNT: {{ 'HPC_ACCOUNT' | getenv }}
BASE_IC: /bucket/Wei.Huang/SFS-REPLAY_ICs/C96mx100
fcst:
TYPE: "hydro"
MONO: "mono"
reforecast: "YES"
FHZER: 12
SFS_POST: "YES"
stage_ic:
USE_OCN_ENS_PERTURB_FILES: "NO"
USE_ATM_ENS_PERTURB_FILES: "NO"
ocn:
MOM6_INTERP_ICS: "YES"
21 changes: 21 additions & 0 deletions sfsc96/sfsc96.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,21 @@
experiment:
system: gefs
mode: forecast-only

arguments:
pslot: {{ 'pslot' | getenv }}
app: S2S
resdetatmos: 96
resensatmos: 96
resdetocean: 1.0
nens: 10
start: cold
comroot: {{ 'RUNTESTS' | getenv }}/COMROOT
expdir: {{ 'RUNTESTS' | getenv }}/EXPDIR
idate: 1994050100
edate: 1994050100
yaml: {{ HOMEgfs }}/sfsc96/options2sfsc96.yaml

skip_ci_on_hosts:
- wcoss2

3 changes: 3 additions & 0 deletions ush/forecast_predet.sh
Original file line number Diff line number Diff line change
Expand Up @@ -559,6 +559,9 @@ FV3_predet(){
${NCP} "${PARMgfs}/post/sfs/postxconfig-NT-sfs.txt" "${DATA}/postxconfig-NT_FH00.txt"
fi
fi

${NCP} "${PARMgfs}/post/sfs/postxconfig-NT-sfs.txt" "${DATA}/postxconfig-NT.txt"
${NCP} "${PARMgfs}/post/sfs/postxconfig-NT-sfs.txt" "${DATA}/postxconfig-NT_FH00.txt"
}

# Disable variable not used warnings
Expand Down
6 changes: 3 additions & 3 deletions versions/build.noaacloud.ver
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
export stack_intel_ver=2021.3.0
export stack_impi_ver=2021.3.0
export stack_intel_ver=2021.10.0
export stack_impi_ver=2021.10.0
export spack_env=gsi-addon-env
source "${HOMEgfs:-}/versions/spack.ver"
export spack_mod_path="/contrib/spack-stack-rocky8/spack-stack-1.6.0/envs/gsi-addon-env/install/modulefiles/Core"
export spack_mod_path="/contrib/Natalie.Perlin/spack-stack/spack-stack-1.6.0/envs/gsi-addon-env/install/modulefiles/Core"
1 change: 1 addition & 0 deletions versions/ic.ver
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,7 @@ ic_versions['C48C48mx500']=20241120
ic_versions['C96']=20241120
ic_versions['C96C48']=20241120
ic_versions['C96mx100']=20240610
ic_versions['C96C96mx100']=20240610

ic_versions['C384C192']=20240610
ic_versions['C384mx025']=20241120
Expand Down
1 change: 1 addition & 0 deletions versions/run.noaacloud.ver
Original file line number Diff line number Diff line change
Expand Up @@ -4,5 +4,6 @@ export spack_env=gsi-addon-env

source "${HOMEgfs:-}/versions/spack.ver"
export spack_mod_path="/contrib/spack-stack-rocky8/spack-stack-1.6.0/envs/gsi-addon-env/install/modulefiles/Core"
export spack_mod_path="/contrib/Natalie.Perlin/spack-stack/spack-stack-1.6.0/envs/gsi-addon-env/install/modulefiles/Core"

export cdo_ver=2.2.0
6 changes: 3 additions & 3 deletions workflow/hosts/awspw.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -19,9 +19,9 @@ CHGRP_CMD: 'chgrp rstprod' # TODO: This is not yet supported.
HPSSARCH: 'NO'
HPSS_PROJECT: emc-global #TODO: See `ATARDIR` below.
BASE_DATA: '/bucket/global-workflow-shared-data'
BASE_IC: '/bucket/global-workflow-shared-data/ICSDIR'
LOCALARCH: 'NO'
ATARDIR: '' # TODO: This will not yet work from AWS.
BASE_IC: '/bucket/Wei.Huang/SFS-REPLAY_ICs/C96mx100'
LOCALARCH: 'YES'
ATARDIR: '/bucket/Wei.Huang/SFS-REPLAY/archive/sfsc96'
MAKE_NSSTBUFR: 'NO'
MAKE_ACFTBUFR: 'NO'
DO_TRACKER: 'NO'
Expand Down

0 comments on commit b6458c7

Please sign in to comment.