add prep_emissions APRUN env. var

NOAA-EPIC · Nov 7, 2024 · 5fb019d · 5fb019d
1 parent 152bb45
commit 5fb019d
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/env/AWSPW.env b/env/AWSPW.env
@@ -13,9 +13,13 @@ export launcher="srun -l --export=ALL"
 export mpmd_opt="--multi-prog --output=mpmd.%j.%t.out"
 
 # Configure MPI environment
-export OMP_STACKSIZE=2048000
+#export OMP_STACKSIZE=2048000
 export NTHSTACK=1024000000
 
+export OMP_STACKSIZE=512M
+export KMP_AFFINITY=scatter
+export OMP_NUM_THREADS=1
+
 ulimit -s unlimited
 ulimit -a
 
@@ -59,6 +63,10 @@ elif [[ "${step}" = "post" ]]; then
     [[ ${NTHREADS_DWN} -gt ${max_threads_per_task} ]] && export NTHREADS_DWN=${max_threads_per_task}
     export APRUN_DWN="${launcher} -n ${ntasks_dwn}"
 
+elif [[ "${step}" = "prep_emissions" ]]; then
+
+    export APRUN="${APRUN_default}"
+
 elif [[ "${step}" = "atmos_products" ]]; then
 
     export USE_CFP="YES"  # Use MPMD for downstream product generation on Hera