aboutsummaryrefslogtreecommitdiff
path: root/examples/gemm_tuner/cl_gemm_reshaped_rhs_only.cpp
diff options
context:
space:
mode:
authorGian Marco Iodice <gianmarco.iodice@arm.com>2021-03-03 17:25:07 +0000
committerGian Marco Iodice <gianmarco.iodice@arm.com>2021-03-04 19:13:40 +0000
commitca419dde35118fcfe07fa0a5ce388c0a40b75c49 (patch)
tree0c57cd91b10d51bfc918edb248319530931aec02 /examples/gemm_tuner/cl_gemm_reshaped_rhs_only.cpp
parent5ff38da7e18e91243a7f6b8e642f8b40f5846068 (diff)
downloadComputeLibrary-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_reshaped_rhs_only.cpp')
-rw-r--r--examples/gemm_tuner/cl_gemm_reshaped_rhs_only.cpp2
1 files changed, 2 insertions, 0 deletions
diff --git a/examples/gemm_tuner/cl_gemm_reshaped_rhs_only.cpp b/examples/gemm_tuner/cl_gemm_reshaped_rhs_only.cpp
index 10fd2984cf..68bec9da6e 100644
--- a/examples/gemm_tuner/cl_gemm_reshaped_rhs_only.cpp
+++ b/examples/gemm_tuner/cl_gemm_reshaped_rhs_only.cpp
@@ -195,6 +195,8 @@ public:
std::cout << "Gemm configurations:" << std::endl;
std::cout << configs << std::endl;
+ tuner.set_tuner_mode(params.tuner_mode);
+
CLScheduler::get().default_init(&tuner);
lhs.allocator()->init(TensorInfo(TensorShape(params.K, params.M, params.B), 1, params.data_type));