Skip to content

Commit

Permalink
Enable MPMD for atmos_products on Hercules. NOAA-EMC#1588
Browse files Browse the repository at this point in the history
  • Loading branch information
DavidHuber-NOAA committed Dec 6, 2023
1 parent 6b1c0af commit 7796272
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 12 deletions.
13 changes: 2 additions & 11 deletions env/HERCULES.env
Original file line number Diff line number Diff line change
Expand Up @@ -62,17 +62,8 @@ elif [[ "${step}" = "upp" ]]; then
[[ ${NTHREADS_UPP} -gt ${nth_max} ]] && export NTHREADS_UPP=${nth_max}
export APRUN_UPP="${launcher} -n ${npe_upp} --cpus-per-task=${NTHREADS_UPP}"

elif [[ "${step}" = "post" ]]; then
elif [[ "${step}" = "atmos_products" ]]; then

nth_max=$((npe_node_max / npe_node_post))

export NTHREADS_NP=${nth_np:-1}
[[ ${NTHREADS_NP} -gt ${nth_max} ]] && export NTHREADS_NP=${nth_max}
export APRUN_NP="${launcher} -n ${npe_post} --cpus-per-task=${NTHREADS_NP}"

export USE_CFP="YES" # Use MPMD for downstream product generation on Hercules
export NTHREADS_DWN=${nth_dwn:-1}
[[ ${NTHREADS_DWN} -gt ${nth_max} ]] && export NTHREADS_DWN=${nth_max}
export APRUN_DWN="${launcher} -n ${npe_dwn} --cpus-per-task=${NTHREADS_DWN}"
export USE_CFP="YES" # Use MPMD for downstream product generation

fi
2 changes: 1 addition & 1 deletion scripts/exglobal_atmos_products.sh
Original file line number Diff line number Diff line change
Expand Up @@ -133,7 +133,7 @@ for (( nset=1 ; nset <= downset ; nset++ )); do
export err=$?
else
chmod 755 "${DATA}/poescript"
bash +x "${DATA}/poescript" 2>&1 mpmd.out
bash +x "${DATA}/poescript" > mpmd.out 2>&1
export err=$?
fi
err_chk
Expand Down

0 comments on commit 7796272

Please sign in to comment.