Skip to content

Commit 38d24ef

Browse files
committed
Allow overriding GPUTYPE/NGPUs from outside
1 parent 4eff422 commit 38d24ef

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

DATA/production/configurations/asyncReco/async_pass.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -414,8 +414,8 @@ echo "SETTING_ROOT_OUTPUT = $SETTING_ROOT_OUTPUT"
414414
# Enabling GPUs
415415
if [[ -n "$ALIEN_JDL_USEGPUS" && $ALIEN_JDL_USEGPUS != 0 ]] ; then
416416
echo "Enabling GPUS"
417-
export GPUTYPE="HIP"
418-
export GPUMEMSIZE=$((25 << 30))
417+
[[ -z ${GPUTYPE:-} ]] && export GPUTYPE="HIP"
418+
[[ -z ${GPUMEMSIZE:-} ]] && export GPUMEMSIZE=$((25 << 30))
419419
if [[ "0$ASYNC_PASS_NO_OPTIMIZED_DEFAULTS" != "01" ]]; then
420420
if [[ "ALIEN_JDL_USEFULLNUMADOMAIN" == 0 ]]; then
421421
if [[ $keep -eq 0 ]]; then

DATA/production/workflow-multiplicities.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -122,15 +122,15 @@ if [[ ! -z ${OPTIMIZED_PARALLEL_ASYNC:-} ]]; then
122122
[[ -z ${TIMEFRAME_RATE_LIMIT:-} ]] && TIMEFRAME_RATE_LIMIT=35
123123
[[ -z ${SHMSIZE:-} ]] && SHMSIZE=100000000000 # SHM_LIMIT 3/4
124124
[[ -z ${TIMEFRAME_SHM_LIMIT:-} ]] && TIMEFRAME_SHM_LIMIT=$(($SHMSIZE / 3))
125+
[[ -z ${NGPUS:-} ]] && NGPUS=4
125126
NGPURECOTHREADS=8
126127
NTRDTRKTHREADS=8
127128
ITSTRK_THREADS=5
128129
ITSTPC_THREADS=3
129130
SVERTEX_THREADS=20
130131
TOFMATCH_THREADS=2
131132
N_SECVTX=2
132-
NGPUS=4
133-
N_TPCTRK=4
133+
N_TPCTRK=$NGPUS
134134
# time in s: pvtx 16, tof 30, trd 82 itstpc 53 its 200 mfttr 30 tpcent 23 hmp-clus 40 (25.11.22)
135135
N_TPCENTDEC=$(math_max $((4 * $NGPUS / 4)) 1)
136136
N_ITSTRK=$(math_max $((12 * $NGPUS / 4)) 1)

0 commit comments

Comments
 (0)