commit ca419dde35118fcfe07fa0a5ce388c0a40b75c49
author Gian Marco Iodice <gianmarco.iodice@arm.com> Wed Mar 03 17:25:07 2021 +0000
committer Gian Marco Iodice <gianmarco.iodice@arm.com> Thu Mar 04 19:13:40 2021 +0000
tree 0c57cd91b10d51bfc918edb248319530931aec02
parent 5ff38da7e18e91243a7f6b8e642f8b40f5846068

Add tuner mode support in GeMM benchmark

Change-Id: Ie1ad9880d22daa2a8ee4f239b5b23d3a7ba4cb1b
Signed-off-by: Gian Marco Iodice <gianmarco.iodice@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/5210
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Reviewed-by: SiCong Li <sicong.li@arm.com>

examples/gemm_tuner/cl_gemm_benchmark.sh

diff --git a/examples/gemm_tuner/cl_gemm_benchmark.sh b/examples/gemm_tuner/cl_gemm_benchmark.sh
index a49c2e1..92fe6b1 100755
--- a/examples/gemm_tuner/cl_gemm_benchmark.sh
+++ b/examples/gemm_tuner/cl_gemm_benchmark.sh
@@ -54,7 +54,10 @@
 DEFAULT_NUM_EXPERIMENTS="all"
 
 # Default output file extension
-DEFAULT_OUT_EXTENSION="gemm_benchmark"
+DEFAULT_OUT_EXTENSION="mlgo_benchmark"
+
+# Default OpenCL tuner mode
+DEFAULT_TUNER_MODE="rapid"
 
 # Number of iterations for each benchmark run
 NUM_ITERATION=5
@@ -269,6 +272,10 @@
         Output file extension.
         Default: ${DEFAULT_OUT_EXTENSION}
 
+        -m <tuner_mode>
+        OpenCL tuner mode.
+        Default: ${DEFAULT_TUNER_MODE}
+
 EOF
 # Print help messages about gemm shapes and various gemm configs
 $HELP && help_gemm_shape_file
@@ -347,6 +354,7 @@
 # Globals:
 #   OUT_DIR
 #   OUT_EXTENSION
+#   TUNER_MODE
 #   EXAMPLE_BIN_DIR
 #   NUM_ITERATION
 #   GEMM_CONFIGS_FILE
@@ -437,7 +445,7 @@
 
     echo "Running shape[$array_shapes_idx]=$gemm_shape with config[$array_configs_idx]=$gemm_config" 1>&2
 
-    example_args="${gemm_shape},${gemm_config},--type=${DATA_TYPE}"
+    example_args="${gemm_shape},${gemm_config},--type=${DATA_TYPE},--tuner-mode=${TUNER_MODE}"
     json_filename="${STRATEGY_OPTION}_${gemm_shape}_${gemm_config}_${DATA_TYPE}"
     # Replace "," with "_"
     json_filename=${json_filename//,/_}
@@ -510,11 +518,13 @@
 NUM_EXPERIMENTS=${DEFAULT_NUM_EXPERIMENTS}
 # Output benchmark result file extension
 OUT_EXTENSION=${DEFAULT_OUT_EXTENSION}
+# OpenCL tuner mode
+TUNER_MODE=${DEFAULT_TUNER_MODE}
 # Toggle help
 HELP=false
 
 # Obtain options
-while getopts "hs:e:g:c:d:o:i:n:t:" opt; do
+while getopts "hs:e:g:c:d:o:i:n:t:m:" opt; do
   case "$opt" in
     h) HELP=true ;;
     s) STRATEGY_OPTION=$(to_lower "${OPTARG}");;
@@ -526,6 +536,7 @@
     i) ID_EXPERIMENT_START="${OPTARG}";;
     n) NUM_EXPERIMENTS="${OPTARG}";;
     t) OUT_EXTENSION="${OPTARG}";;
+    m) TUNER_MODE="${OPTARG}";;
   esac
 done
 shift $((OPTIND - 1))