aboutsummaryrefslogtreecommitdiff
path: root/src/core/CL/kernels/CLWeightsReshapeKernel.cpp
diff options
context:
space:
mode:
authorGian Marco Iodice <gianmarco.iodice@arm.com>2017-07-25 10:25:53 +0100
committerAnthony Barbier <anthony.barbier@arm.com>2018-09-17 14:16:42 +0100
commit3a62324f8f0a35c6f2c69cbc38cc1d52863c4ba8 (patch)
tree365c2f686f40dc1cd1e3826915db581562ca2605 /src/core/CL/kernels/CLWeightsReshapeKernel.cpp
parentb9e9cff3a9293c60b53683ccfbf2c50dbbbbf44e (diff)
downloadComputeLibrary-3a62324f8f0a35c6f2c69cbc38cc1d52863c4ba8.tar.gz
COMPMID-455 - Optimizing CLIm2ColKernel
Change-Id: Iee618948cc8f310ee9af2d786240e8120e4c6ab9 Reviewed-on: http://mpd-gerrit.cambridge.arm.com/81665 Tested-by: Kaizen <jeremy.johnson+kaizengerrit@arm.com> Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Diffstat (limited to 'src/core/CL/kernels/CLWeightsReshapeKernel.cpp')
-rw-r--r--src/core/CL/kernels/CLWeightsReshapeKernel.cpp4
1 files changed, 4 insertions, 0 deletions
diff --git a/src/core/CL/kernels/CLWeightsReshapeKernel.cpp b/src/core/CL/kernels/CLWeightsReshapeKernel.cpp
index b802c862fc..7b80f3ff5a 100644
--- a/src/core/CL/kernels/CLWeightsReshapeKernel.cpp
+++ b/src/core/CL/kernels/CLWeightsReshapeKernel.cpp
@@ -78,6 +78,10 @@ void CLWeightsReshapeKernel::configure(const ICLTensor *input, const ICLTensor *
std::set<std::string> build_opts;
build_opts.emplace(("-DDATA_TYPE=" + get_cl_type_from_data_type(input->info()->data_type())));
build_opts.emplace(((biases != nullptr) ? "-DHAS_BIAS" : ""));
+ if(is_data_type_fixed_point(input->info()->data_type()))
+ {
+ build_opts.emplace("-DFIXED_POINT_POSITION=" + support::cpp11::to_string(input->info()->fixed_point_position()));
+ }
// Create kernel
_kernel = static_cast<cl::Kernel>(CLKernelLibrary::get().create_kernel("reshape_to_columns", build_opts));