Add a `--iterations` option for `benchmarks_run_target.sh`.

Change-Id: Ia445a9cdbf0621d81edede695902cb1b42d7bacc

1 files changed, 15 insertions, 3 deletions

diff --git a/benchmarks/benchmarks_run_target.sh b/benchmarks/benchmarks_run_target.sh
index 27b6d7bf..35479efc 100755
--- a/benchmarks/benchmarks_run_target.sh
+++ b/benchmarks/benchmarks_run_target.sh
@@ -10,17 +10,19 @@ readonly log_directory="$(get_workspace)"
 readonly test_time_file="${log_directory}/time_benchmarks.txt"
 
 declare -A options
-readonly iterations=10
+readonly default_iterations=10
 tests_summary=
 
 init_options() {
   options["mode"]="all"
   options["cpu"]="all"
+  options["iterations"]=$default_iterations
 }
 
 set_default_options() {
   options["mode"]="all"
   options["cpu"]="all"
+  options["iterations"]=$default_iterations
 }
 
 validate_options() {
@@ -32,6 +34,10 @@ validate_options() {
   if [[ "$cpu" != "all" && "$cpu" != "big" && "$cpu" != "little" ]]; then
     log E "Invalid cpu option: $cpu"
   fi
+  local -r iterations="${options["iterations"]}"
+  if [[ ! ${iterations} =~ ^[0-9]+$ ]]; then
+    log E "Invalid number of iterations: ${iterations}"
+  fi
 }
 
 dump_options() {
@@ -55,10 +61,12 @@ usage() {
   log I "                          \`big\` and \`little\` values are only valid"
   log I "                          for big.LITTLE devices."
   log I "                          (default: all)"
+  log I " --iterations <n>       - The number of iterations to run the benchmarks for."
+  log I "                          (default: $default_iterations)"
   log I "-------------------------------------------"
   log I "Default Configuration:"
   log I " --default              - Default benchmark configuration, equivalent to"
-  log I "                          \`--mode all --cpu all\`."
+  log I "                          \`--mode all --cpu all --iterations $default_iterations\`."
   log I "-------------------------------------------"
 }
 
@@ -82,6 +90,10 @@ arguments_parser() {
         shift
         options["cpu"]=$1
         ;;
+      --iterations)
+        shift
+        options["iterations"]=$1
+        ;;
       # Default:
       --default)
         set_default_options
@@ -180,7 +192,7 @@ sync_target() {
 
 run_benchmarks() {
   start_section "run_benchmarks_$2_$1_$3"
-  safe ./benchmarks/run.py --target --iterations "${iterations}" --mode "$1" \
+  safe ./benchmarks/run.py --target --iterations "${options["iterations"]}" --mode "$1" \
   --android-root "${ART_TEST_ANDROID_ROOT}" --output-json "${log_directory}/$2_$1_$3_aot.json"
   end_section "run_benchmarks_$2_$1_$3"
 }