update env

ypwang19 · May 30, 2024 · 18f11fb · 18f11fb
1 parent 68d3e2c
commit 18f11fb
Show file tree

Hide file tree

Showing 5 changed files with 35 additions and 0 deletions.
diff --git 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
@@ -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
@@ -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
@@ -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
@@ -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))