Skip to content

Commit ce29449

Browse files
committed
Async reco settings for NERSC cluster
1 parent bc937bf commit ce29449

File tree

1 file changed

+14
-4
lines changed

1 file changed

+14
-4
lines changed

DATA/production/workflow-multiplicities.sh

Lines changed: 14 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -91,7 +91,7 @@ if [[ ! -z ${OPTIMIZED_PARALLEL_ASYNC:-} ]]; then
9191
NTRDTRKTHREADS=3
9292
ITSTRK_THREADS=2
9393
ITSTPC_THREADS=2
94-
elif [[ $OPTIMIZED_PARALLEL_ASYNC == "pp_4gpu" ]]; then
94+
elif [[ $OPTIMIZED_PARALLEL_ASYNC =~ ^pp_4gpu(_|$) ]]; then
9595
if [[ -z ${TIMEFRAME_RATE_LIMIT:-} ]]; then
9696
if [[ ! -z ${ALIEN_JDL_LPMANCHORYEAR} && ${ALIEN_JDL_LPMANCHORYEAR} -lt 2023 ]]; then
9797
TIMEFRAME_RATE_LIMIT=45
@@ -100,13 +100,18 @@ if [[ ! -z ${OPTIMIZED_PARALLEL_ASYNC:-} ]]; then
100100
fi
101101
fi
102102
[[ -z ${SHMSIZE:-} ]] && SHMSIZE=100000000000
103+
if [[ $OPTIMIZED_PARALLEL_ASYNC == "pp_4gpu_NERSC" ]]; then
104+
NGPUS=1
105+
GPUTYPE=CUDA
106+
else
107+
NGPUS=4
108+
fi
103109
NGPURECOTHREADS=8
104110
NTRDTRKTHREADS=2
105111
ITSTRK_THREADS=2
106112
ITSTPC_THREADS=2
107113
SVERTEX_THREADS=4
108114
TPCTIMESERIES_THREADS=2
109-
NGPUS=4
110115
N_TPCTRK=4
111116
N_FWDMATCH=2
112117
N_PRIMVTXMATCH=1
@@ -123,11 +128,16 @@ if [[ ! -z ${OPTIMIZED_PARALLEL_ASYNC:-} ]]; then
123128
N_ITSTRK=12
124129
N_ITSCL=2
125130
export DPL_SMOOTH_RATE_LIMITING=1
126-
elif [[ $OPTIMIZED_PARALLEL_ASYNC == "PbPb_4gpu" ]]; then
131+
elif [[ $OPTIMIZED_PARALLEL_ASYNC =~ ^PbPb_4gpu(_|$) ]]; then
127132
[[ -z ${TIMEFRAME_RATE_LIMIT:-} ]] && TIMEFRAME_RATE_LIMIT=35
128133
[[ -z ${SHMSIZE:-} ]] && SHMSIZE=100000000000 # SHM_LIMIT 3/4
129134
[[ -z ${TIMEFRAME_SHM_LIMIT:-} ]] && TIMEFRAME_SHM_LIMIT=$(($SHMSIZE / 3))
130-
NGPUS=4
135+
if [[ $OPTIMIZED_PARALLEL_ASYNC == "PbPb_4gpu_NERSC" ]]; then
136+
NGPUS=1
137+
GPUTYPE=CUDA
138+
else
139+
NGPUS=4
140+
fi
131141
NGPURECOTHREADS=8
132142
NTRDTRKTHREADS=8
133143
ITSTRK_THREADS=5

0 commit comments

Comments
 (0)