|
20 | 20 | # Benchmark defaults. All can be overridden by exporting variables before calling this script. |
21 | 21 |
|
22 | 22 | export DPL_REPORT_PROCESSING="${DPL_REPORT_PROCESSING:-1}" |
23 | | - |
24 | | -# export FST_TMUX_NO_EPN="${FST_TMUX_NO_EPN:-1}" |
25 | 23 | export WORKFLOW_PARAMETERS="${WORKFLOW_PARAMETERS:-GPU,CTF}" |
26 | 24 | export GPUTYPE="${GPUTYPE:-CUDA}" |
27 | 25 | export NGPUS=1 |
28 | | - |
29 | 26 | export O2_GPU_DOUBLE_PIPELINE="${O2_GPU_DOUBLE_PIPELINE:-1}" |
30 | 27 | export O2_GPU_RTC="${O2_GPU_RTC:-1}" |
31 | | - |
32 | | -export EPNSYNCMODE="${EPNSYNCMODE:-0}" |
33 | 28 | export SYNCMODE="${SYNCMODE:-1}" |
34 | | -export SYNCRAWMODE="${SYNCRAWMODE:-0}" |
35 | | - |
36 | | -export TIMEFRAME_RATE_LIMIT="${TIMEFRAME_RATE_LIMIT:-5}" |
37 | | - |
38 | 29 | export DISABLE_ROOT_OUTPUT="${DISABLE_ROOT_OUTPUT:-1}" |
39 | 30 |
|
40 | 31 | # Double pipeline requires zsraw input. Therefore default to raw TF input, not CTF. |
41 | | -export CTFINPUT="${CTFINPUT:-0}" |
42 | 32 | export RAWTFINPUT="${RAWTFINPUT:-1}" |
43 | | -export DIGITINPUT="${DIGITINPUT:-0}" |
44 | | -export EXTINPUT="${EXTINPUT:-0}" |
45 | 33 |
|
46 | 34 | export NTIMEFRAMES="${NTIMEFRAMES:--1}" |
47 | 35 | export TFLOOP="${TFLOOP:-100}" |
48 | 36 | export TFDELAY="${TFDELAY:-0}" |
| 37 | +export TIMEFRAME_RATE_LIMIT="${TIMEFRAME_RATE_LIMIT:-5}" |
49 | 38 |
|
50 | 39 | export RUN_BENCHMARK="${RUN_BENCHMARK:-0}" |
51 | 40 |
|
52 | | -if [[ -f "$PWD/local_env.sh" ]]; then |
53 | | - source "$PWD/local_env.sh" |
54 | | -fi |
55 | | - |
56 | 41 | echo "# Alien/JAliEn environment check:" |
57 | 42 | echo "# JALIEN_TOKEN_CERT=${JALIEN_TOKEN_CERT:-}" |
58 | 43 | echo "# JALIEN_TOKEN_KEY=${JALIEN_TOKEN_KEY:-}" |
@@ -178,36 +163,6 @@ if [[ -n "${BENCH_EXTRA_LD_LIBRARY_PATH:-}" ]]; then |
178 | 163 | export LD_LIBRARY_PATH="$BENCH_EXTRA_LD_LIBRARY_PATH:${LD_LIBRARY_PATH:-}" |
179 | 164 | fi |
180 | 165 |
|
181 | | -# Fail early for the unsupported combination instead of letting o2-gpu-reco-workflow crash later. |
182 | | -if [[ "${O2_GPU_DOUBLE_PIPELINE:-0}" == "1" ]]; then |
183 | | - if [[ "${CTFINPUT:-0}" == "1" ]]; then |
184 | | - echo "FATAL: O2_GPU_DOUBLE_PIPELINE=1 is incompatible with CTFINPUT=1 in dpl-workflow.sh." >&2 |
185 | | - echo "Double pipeline requires o2-gpu-reco-workflow --input-type=zsraw." >&2 |
186 | | - echo "Use RAWTFINPUT=1 or rawAll.cfg input, or set O2_GPU_DOUBLE_PIPELINE=0 for CTF benchmarking." >&2 |
187 | | - exit 1 |
188 | | - fi |
189 | | - if [[ "${DIGITINPUT:-0}" == "1" ]]; then |
190 | | - echo "FATAL: O2_GPU_DOUBLE_PIPELINE=1 is not suitable for DIGITINPUT=1 in dpl-workflow.sh." >&2 |
191 | | - echo "Digit input uses zsonthefly and is restricted to NTIMEFRAMES=1." >&2 |
192 | | - exit 1 |
193 | | - fi |
194 | | -fi |
195 | | - |
196 | | -# Input checks with clearer messages. |
197 | | -if [[ "${RAWTFINPUT:-0}" == "1" ]]; then |
198 | | - if [[ -z "${FILEWORKDIR:-}" && -z "${INPUT_FILE_LIST:-}" ]]; then |
199 | | - echo "FATAL: RAWTFINPUT=1 but neither FILEWORKDIR nor INPUT_FILE_LIST is set." >&2 |
200 | | - echo "Set FILEWORKDIR=/path/to/raw_tf_dir or INPUT_FILE_LIST=/path/to/o2_*.tf" >&2 |
201 | | - exit 1 |
202 | | - fi |
203 | | - if [[ -z "${INPUT_FILE_LIST:-}" ]] && ! ls "${FILEWORKDIR}"/o2_*.tf >/dev/null 2>&1; then |
204 | | - echo "FATAL: RAWTFINPUT=1 but no raw TF file was found." >&2 |
205 | | - echo "Looked for: ${FILEWORKDIR}/o2_*.tf" >&2 |
206 | | - echo "Set FILEWORKDIR=/path/to/raw_tf_dir or INPUT_FILE_LIST=/path/to/o2_*.tf" >&2 |
207 | | - exit 1 |
208 | | - fi |
209 | | -fi |
210 | | - |
211 | 166 | # Check CUDA runtime/device visibility before starting the full workflow. |
212 | 167 | if [[ "$GPUTYPE" == "CUDA" ]]; then |
213 | 168 | if ! command -v nvidia-smi >/dev/null 2>&1; then |
@@ -256,7 +211,7 @@ echo "# output dir: $OUTDIR" |
256 | 211 | echo "# run dir: $RUNDIR" |
257 | 212 | echo "# NGPUS=$NGPUS GPUTYPE=$GPUTYPE" |
258 | 213 | echo "# O2_GPU_DOUBLE_PIPELINE=$O2_GPU_DOUBLE_PIPELINE O2_GPU_RTC=$O2_GPU_RTC" |
259 | | -echo "# CTFINPUT=$CTFINPUT RAWTFINPUT=$RAWTFINPUT DIGITINPUT=$DIGITINPUT EXTINPUT=$EXTINPUT NTIMEFRAMES=$NTIMEFRAMES TFLOOP=$TFLOOP" |
| 214 | +echo "# NTIMEFRAMES=$NTIMEFRAMES TFLOOP=$TFLOOP" |
260 | 215 | echo "# FILEWORKDIR=${FILEWORKDIR:-} INPUT_FILE_LIST=${INPUT_FILE_LIST:-}" |
261 | 216 | echo "# LD_LIBRARY_PATH font-lib workaround: BENCH_USE_SYSTEM_FONT_LIBS=$BENCH_USE_SYSTEM_FONT_LIBS" |
262 | 217 | echo "# ROCm library auto-detect: BENCH_AUTO_ROCM_LIBS=$BENCH_AUTO_ROCM_LIBS (active only when GPUTYPE=HIP)" |
|
0 commit comments