1919# ----------------------------------------------------------------------------------------------------------------------
2020# Benchmark defaults. All can be overridden by exporting variables before calling this script.
2121
22+ case " ${GPUTYPE:- } " in
23+ CUDA|HIP)
24+ export GPUTYPE
25+ ;;
26+ " " )
27+ echo " ERROR: GPUTYPE must be set to either CUDA or HIP" >&2
28+ exit 1
29+ ;;
30+ * )
31+ echo " ERROR: Invalid GPUTYPE='$GPUTYPE '. Must be either CUDA or HIP" >&2
32+ exit 1
33+ ;;
34+ esac
35+
2236export DPL_REPORT_PROCESSING=" ${DPL_REPORT_PROCESSING:- 1} "
2337export WORKFLOW_PARAMETERS=" ${WORKFLOW_PARAMETERS:- GPU,CTF} "
24- export GPUTYPE=" ${GPUTYPE:- CUDA} "
2538export NGPUS=1
2639export O2_GPU_DOUBLE_PIPELINE=" ${O2_GPU_DOUBLE_PIPELINE:- 1} "
2740export O2_GPU_RTC=" ${O2_GPU_RTC:- 1} "
2841export SYNCMODE=" ${SYNCMODE:- 1} "
29- export DISABLE_ROOT_OUTPUT=" ${DISABLE_ROOT_OUTPUT:- 1} "
3042
3143# Double pipeline requires zsraw input. Therefore default to raw TF input, not CTF.
3244export RAWTFINPUT=" ${RAWTFINPUT:- 1} "
@@ -35,67 +47,10 @@ export NTIMEFRAMES="${NTIMEFRAMES:--1}"
3547export TFLOOP=" ${TFLOOP:- 100} "
3648export TFDELAY=" ${TFDELAY:- 0} "
3749export TIMEFRAME_RATE_LIMIT=" ${TIMEFRAME_RATE_LIMIT:- 5} "
50+ export ARGS_EXTRA_PROCESS_o2_gpu_reco_workflow=" ${ARGS_EXTRA_PROCESS_o2_gpu_reco_workflow: +$ARGS_EXTRA_PROCESS_o2_gpu_reco_workflow } --log-timestamp-us"
3851
3952export RUN_BENCHMARK=" ${RUN_BENCHMARK:- 0} "
4053
41- echo " # Alien/JAliEn environment check:"
42- echo " # JALIEN_TOKEN_CERT=${JALIEN_TOKEN_CERT:- } "
43- echo " # JALIEN_TOKEN_KEY=${JALIEN_TOKEN_KEY:- } "
44- echo " # X509_USER_PROXY=${X509_USER_PROXY:- } "
45- if command -v alien-token-info > /dev/null 2>&1 ; then
46- alien-token-info || true
47- else
48- echo " # alien-token-info not found in PATH"
49- fi
50- echo
51-
52- # ----------------------------------------------------------------------------------------------------------------------
53- # Recover JAliEn token environment if alien-token-info works but token env vars are missing.
54-
55- if command -v alien-token-info > /dev/null 2>&1 ; then
56- if alien-token-info > /dev/null 2>&1 ; then
57- uid=" $( id -u) "
58-
59- for cert in \
60- " /tmp/jalien_token_${uid} .pem" \
61- " /tmp/jalien_token_${USER} .pem" \
62- " /tmp/tokencert_${uid} .pem" \
63- " /tmp/tokencert_${USER} .pem"
64- do
65- if [[ -f " $cert " ]]; then
66- export JALIEN_TOKEN_CERT=" ${JALIEN_TOKEN_CERT:- $cert } "
67- break
68- fi
69- done
70-
71- for key in \
72- " /tmp/jalien_token_${uid} .key" \
73- " /tmp/jalien_token_${USER} .key" \
74- " /tmp/tokenkey_${uid} .pem" \
75- " /tmp/tokenkey_${USER} .pem"
76- do
77- if [[ -f " $key " ]]; then
78- export JALIEN_TOKEN_KEY=" ${JALIEN_TOKEN_KEY:- $key } "
79- break
80- fi
81- done
82-
83- # Some older tools only look for X509_USER_PROXY.
84- if [[ -z " ${X509_USER_PROXY:- } " ]]; then
85- for proxy in \
86- " /tmp/x509up_u${uid} " \
87- " /tmp/x509up_${uid} " \
88- " ${JALIEN_TOKEN_CERT:- } "
89- do
90- if [[ -n " $proxy " && -f " $proxy " ]]; then
91- export X509_USER_PROXY=" $proxy "
92- break
93- fi
94- done
95- fi
96- fi
97- fi
98-
9954# ----------------------------------------------------------------------------------------------------------------------
10055# Benchmark naming / output directory.
10156
@@ -115,9 +70,6 @@ cleanup_rundir() {
11570
11671trap cleanup_rundir EXIT
11772
118- # ----------------------------------------------------------------------------------------------------------------------
119- # Keep accidental files out of the source/original directory.
120-
12173# Let O2/core dumps land in the benchmark run directory, not in the original working directory.
12274export CORE_DUMP_DIR=" ${CORE_DUMP_DIR:- $RUNDIR } "
12375export O2_CORE_DUMP_DIR=" ${O2_CORE_DUMP_DIR:- $RUNDIR } "
0 commit comments