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_reshaped.cpp | |
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_reshaped.cpp')
-rw-r--r-- | examples/gemm_tuner/cl_gemm_reshaped.cpp | 2 |
1 files changed, 2 insertions, 0 deletions
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)); |