diff options
author | Gian Marco Iodice <gianmarco.iodice@arm.com> | 2021-03-03 17:25:07 +0000 |
---|---|---|
committer | Gian Marco Iodice <gianmarco.iodice@arm.com> | 2021-03-04 19:13:40 +0000 |
commit | ca419dde35118fcfe07fa0a5ce388c0a40b75c49 (patch) | |
tree | 0c57cd91b10d51bfc918edb248319530931aec02 /examples/gemm_tuner/cl_gemm_benchmark.sh | |
parent | 5ff38da7e18e91243a7f6b8e642f8b40f5846068 (diff) | |
download | ComputeLibrary-ca419dde35118fcfe07fa0a5ce388c0a40b75c49.tar.gz |
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>
Diffstat (limited to 'examples/gemm_tuner/cl_gemm_benchmark.sh')
-rwxr-xr-x | examples/gemm_tuner/cl_gemm_benchmark.sh | 17 |
1 files changed, 14 insertions, 3 deletions
diff --git a/examples/gemm_tuner/cl_gemm_benchmark.sh b/examples/gemm_tuner/cl_gemm_benchmark.sh index a49c2e154b..92fe6b194e 100755 --- a/examples/gemm_tuner/cl_gemm_benchmark.sh +++ b/examples/gemm_tuner/cl_gemm_benchmark.sh @@ -54,7 +54,10 @@ DEFAULT_ID_EXPERIMENT_START=0 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 @@ Options: 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 @@ function arr_contains() { # Globals: # OUT_DIR # OUT_EXTENSION +# TUNER_MODE # EXAMPLE_BIN_DIR # NUM_ITERATION # GEMM_CONFIGS_FILE @@ -437,7 +445,7 @@ function run() { 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 @@ ID_EXPERIMENT_START=${DEFAULT_ID_EXPERIMENT_START} 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 @@ while getopts "hs:e:g:c:d:o:i:n:t:" opt; do i) ID_EXPERIMENT_START="${OPTARG}";; n) NUM_EXPERIMENTS="${OPTARG}";; t) OUT_EXTENSION="${OPTARG}";; + m) TUNER_MODE="${OPTARG}";; esac done shift $((OPTIND - 1)) |