-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathexec_bench.sh
executable file
·47 lines (34 loc) · 1.23 KB
/
exec_bench.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
#!/bin/sh
# This shell script is intended to be executed into the docker container
# Configuration
BATCHES_PER_RUN=$1
CPU_MODEL=$2
GPU_MODEL=$3
POWER_LIMIT=$4
ITERATIONS=$5
# CONFIG=config/config_all
# CONFIG=config/config_resnet50_replicated_fp16_train_syn
CONFIG=config/tune
cd lambda-tensorflow-benchmark
echo
echo "**** Executing tests with BATCHES_PER_RUN="$BATCHES_PER_RUN", POWER_LIMIT="$POWER_LIMIT"W, ITERATIONS="$ITERATIONS" and CONFIG="$CONFIG"..."
echo
TF_FORCE_GPU_ALLOW_GROWTH=true TF_XLA_FLAGS=--tf_xla_auto_jit=2 ./batch_benchmark.sh 1 1 $ITERATIONS $BATCHES_PER_RUN 2 $CONFIG
LOGS_PATH="logs/"$CPU_MODEL"-"$GPU_MODEL".logs"
REPORT_PATH="../results/"$CPU_MODEL"-"$GPU_MODEL".md"
echo
echo "Generating reports..."
echo "Logs path: $LOGS_PATH"
echo "Report path: $REPORT_PATH"
echo
../subreport_add.sh $LOGS_PATH $POWER_LIMIT $BATCHES_PER_RUN $ITERATIONS >> $REPORT_PATH
EXPORT_LOGS_PATH="../results/"$CPU_MODEL"-"$GPU_MODEL"-"$POWER_LIMIT"W.logs"
echo
echo "Archiving logs"
echo "Export log path: $EXPORT_LOGS_PATH"
echo
rm -rf $EXPORT_LOGS_PATH
mv $LOGS_PATH $EXPORT_LOGS_PATH
echo
echo "**** Test ENDED for BATCHES_PER_RUN="$BATCHES_PER_RUN", POWER_LIMIT="$POWER_LIMIT"W, ITERATIONS="$ITERATIONS" and CONFIG="$CONFIG"..."
echo