From 18f11fb7dac605ddb5a0b2ac04c115de91d8abb2 Mon Sep 17 00:00:00 2001
From: ypwang19 <yaping.wang@users.noreply.github.com>
Date: Thu, 30 May 2024 13:07:32 +0000
Subject: [PATCH] update env

---
 env/HERCULES.env | 7 +++++++
 env/JET.env      | 7 +++++++
 env/ORION.env    | 7 +++++++
 env/S4.env       | 7 +++++++
 env/WCOSS2.env   | 7 +++++++
 5 files changed, 35 insertions(+)

diff --git a/env/HERCULES.env b/env/HERCULES.env
index 0824ba913a..6067bb3de3 100755
--- a/env/HERCULES.env
+++ b/env/HERCULES.env
@@ -99,6 +99,13 @@ case ${step} in
     [[ ${NTHREADS_AEROANL} -gt ${nth_max} ]] && export NTHREADS_AEROANL=${nth_max}
     export APRUN_AEROANL="${launcher} -n ${npe_aeroanlrun} --cpus-per-task=${NTHREADS_AEROANL}"
  ;;
+
+ "prepaeroobs")
+    nth_max=$((npe_node_max / npe_node_prepaeroobs))
+
+    export NTHREADS_PREPAEROOBS=${nth_prepaeroobs:-1}
+    export APRUN_PREPAEROOBS="${launcher} -n 1 --cpus-per-task=${NTHREADS_PREPAEROOBS}"
+
  "snowanl")
 
     nth_max=$((npe_node_max / npe_node_snowanl))
diff --git a/env/JET.env b/env/JET.env
index 5bd88dc93a..900ab9920b 100755
--- a/env/JET.env
+++ b/env/JET.env
@@ -82,6 +82,13 @@ elif [[ "${step}" = "aeroanlrun" ]]; then
     [[ ${NTHREADS_AEROANL} -gt ${nth_max} ]] && export NTHREADS_AEROANL=${nth_max}
     export APRUN_AEROANL="${launcher} -n ${npe_aeroanlrun}"
 
+elif [[ "${step}" = "prepaeroobs" ]]; then
+
+    nth_max=$((npe_node_max / npe_node_prepaeroobs))
+
+    export NTHREADS_PREPAEROOBS=${nth_prepaeroobs:-1}
+    export APRUN_PREPAEROOBS="${launcher} -n 1 --cpus-per-task=${NTHREADS_PREPAEROOBS}"
+
 elif [[ "${step}" = "snowanl" ]]; then
 
     nth_max=$((npe_node_max / npe_node_snowanl))
diff --git a/env/ORION.env b/env/ORION.env
index f701e55aa2..9da4545cc2 100755
--- a/env/ORION.env
+++ b/env/ORION.env
@@ -90,6 +90,13 @@ elif [[ "${step}" = "aeroanlrun" ]]; then
     [[ ${NTHREADS_AEROANL} -gt ${nth_max} ]] && export NTHREADS_AEROANL=${nth_max}
     export APRUN_AEROANL="${launcher} -n ${npe_aeroanlrun} --cpus-per-task=${NTHREADS_AEROANL}"
 
+elif [[ "${step}" = "prepaeroobs" ]]; then
+
+    nth_max=$((npe_node_max / npe_node_prepaeroobs))
+
+    export NTHREADS_PREPAEROOBS=${nth_prepaeroobs:-1}
+    export APRUN_PREPAEROOBS="${launcher} -n 1 --cpus-per-task=${NTHREADS_PREPAEROOBS}"
+
 elif [[ "${step}" = "snowanl" ]]; then
 
     nth_max=$((npe_node_max / npe_node_snowanl))
diff --git a/env/S4.env b/env/S4.env
index 9ba3a61b01..8704bcec0e 100755
--- a/env/S4.env
+++ b/env/S4.env
@@ -82,6 +82,13 @@ elif [[ "${step}" = "aeroanlrun" ]]; then
     [[ ${NTHREADS_AEROANL} -gt ${nth_max} ]] && export NTHREADS_AEROANL=${nth_max}
     export APRUN_AEROANL="${launcher} -n ${npe_aeroanlrun}"
 
+elif [[ "${step}" = "prepaeroobs" ]]; then
+
+    nth_max=$((npe_node_max / npe_node_prepaeroobs))
+
+    export NTHREADS_PREPAEROOBS=${nth_prepaeroobs:-1}
+    export APRUN_PREPAEROOBS="${launcher} -n 1 --cpus-per-task=${NTHREADS_PREPAEROOBS}"
+
 elif [[ "${step}" = "snowanl" ]]; then
 
     nth_max=$((npe_node_max / npe_node_snowanl))
diff --git a/env/WCOSS2.env b/env/WCOSS2.env
index 0876e4127d..2e2daaf540 100755
--- a/env/WCOSS2.env
+++ b/env/WCOSS2.env
@@ -76,6 +76,13 @@ elif [[ "${step}" = "aeroanlrun" ]]; then
     [[ ${NTHREADS_AEROANL} -gt ${nth_max} ]] && export NTHREADS_AEROANL=${nth_max}
     export APRUN_AEROANL="${launcher} -n ${npe_aeroanlrun}"
 
+elif [[ "${step}" = "prepaeroobs" ]]; then
+
+    nth_max=$((npe_node_max / npe_node_prepaeroobs))
+
+    export NTHREADS_PREPAEROOBS=${nth_prepaeroobs:-1}
+    export APRUN_PREPAEROOBS="${launcher} -n 1 --cpus-per-task=${NTHREADS_PREPAEROOBS}"
+
 elif [[ "${step}" = "snowanl" ]]; then
 
     nth_max=$((npe_node_max / npe_node_snowanl))