From ca419dde35118fcfe07fa0a5ce388c0a40b75c49 Mon Sep 17 00:00:00 2001 From: Gian Marco Iodice Date: Wed, 3 Mar 2021 17:25:07 +0000 Subject: Add tuner mode support in GeMM benchmark Change-Id: Ie1ad9880d22daa2a8ee4f239b5b23d3a7ba4cb1b Signed-off-by: Gian Marco Iodice Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/5210 Tested-by: Arm Jenkins Reviewed-by: Georgios Pinitas Reviewed-by: SiCong Li --- examples/gemm_tuner/cl_gemm_reshaped.cpp | 2 ++ 1 file changed, 2 insertions(+) (limited to 'examples/gemm_tuner/cl_gemm_reshaped.cpp') diff --git a/examples/gemm_tuner/cl_gemm_reshaped.cpp b/examples/gemm_tuner/cl_gemm_reshaped.cpp index 2ea4769dd3..444a342d74 100644 --- a/examples/gemm_tuner/cl_gemm_reshaped.cpp +++ b/examples/gemm_tuner/cl_gemm_reshaped.cpp @@ -222,6 +222,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)); -- cgit v1.2.1