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_native.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_native.cpp')
-rw-r--r-- | examples/gemm_tuner/cl_gemm_native.cpp | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/examples/gemm_tuner/cl_gemm_native.cpp b/examples/gemm_tuner/cl_gemm_native.cpp index 02f144ea12..5a144dabf7 100644 --- a/examples/gemm_tuner/cl_gemm_native.cpp +++ b/examples/gemm_tuner/cl_gemm_native.cpp @@ -1,5 +1,5 @@ /* - * Copyright (c) 2019-2020 Arm Limited. + * Copyright (c) 2019-2021 Arm Limited. * * SPDX-License-Identifier: MIT * @@ -170,6 +170,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)); |