diff options
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)); |