Skip to content

Commit 5d12c67

Browse files
committed
Minor polishing
1 parent 013c66d commit 5d12c67

1 file changed

Lines changed: 11 additions & 4 deletions

File tree

prodtests/full-system-test/gen_single_gpu_rtc_benchmark.sh

Lines changed: 11 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -16,9 +16,6 @@ if [[ ! -f "$O2_DPL_WORKFLOW" ]]; then
1616
exit 1
1717
fi
1818

19-
# ----------------------------------------------------------------------------------------------------------------------
20-
# Benchmark defaults. All can be overridden by exporting variables before calling this script.
21-
2219
case "${GPUTYPE:-}" in
2320
CUDA|HIP|CPU|OPENCL)
2421
export GPUTYPE
@@ -33,6 +30,14 @@ case "${GPUTYPE:-}" in
3330
;;
3431
esac
3532

33+
if [[ -z "${FILEWORKDIR:-}" ]]; then
34+
echo "ERROR: FILEWORKDIR must be set to a directory containing raw TF input files" >&2
35+
exit 1
36+
fi
37+
38+
# ----------------------------------------------------------------------------------------------------------------------
39+
# Benchmark defaults. All can be overridden by exporting variables before calling this script.
40+
3641
export DPL_REPORT_PROCESSING="${DPL_REPORT_PROCESSING:-1}"
3742
export WORKFLOW_PARAMETERS="${WORKFLOW_PARAMETERS:-GPU,CTF}"
3843
export NGPUS="${NGPUS:-1}"
@@ -73,6 +78,7 @@ cleanup_rundir() {
7378
echo "# Cleaning run dir: $RUNDIR"
7479
rm -rf -- "$RUNDIR"
7580
fi
81+
rm -rf /dev/shm/fmq*
7682
}
7783

7884
trap cleanup_rundir EXIT
@@ -97,10 +103,11 @@ echo "# single-GPU RTC benchmark"
97103
echo "# source script: $O2_DPL_WORKFLOW"
98104
echo "# output dir: $OUTDIR"
99105
echo "# run dir: $RUNDIR"
106+
echo "# run benchmark: $RUN_BENCHMARK (0: prints workflow, 1: runs workflow)"
100107
echo "# NGPUS=$NGPUS GPUTYPE=$GPUTYPE"
101108
echo "# O2_GPU_DOUBLE_PIPELINE=$O2_GPU_DOUBLE_PIPELINE O2_GPU_RTC=$O2_GPU_RTC"
102109
echo "# NTIMEFRAMES=$NTIMEFRAMES TFLOOP=$TFLOOP"
103-
echo "# FILEWORKDIR=${FILEWORKDIR:-} INPUT_FILE_LIST=${INPUT_FILE_LIST:-}"
110+
echo "# FILEWORKDIR=${FILEWORKDIR:-}
104111
echo "# LD_LIBRARY_PATH is not modified by this script"
105112
echo
106113

0 commit comments

Comments
 (0)