diff options
| author | Gian Marco Iodice <gianmarco.iodice@arm.com> | 2018-12-06 17:13:09 +0000 |
|---|---|---|
| committer | Gian Marco Iodice <gianmarco.iodice@arm.com> | 2018-12-10 15:58:54 +0000 |
| commit | 5ba5e0938e68d4f90f5545a81066d56f022b376a (patch) | |
| tree | d828f8b3fd52e6d5b8f732a7ec41f832f0b921d8 /src/core/CL/kernels/CLReverseKernel.cpp | |
| parent | 1d7cbb99d2a34abd15f3b6c2e017115736cd90cc (diff) | |
| download | ComputeLibrary-5ba5e0938e68d4f90f5545a81066d56f022b376a.tar.gz | |
COMPMID-1774: Implement CLGEMMReshapeLHSMatrixKernel to reshape the LHS matrix of GEMM/GEMMLowp
Change-Id: I8c5fd4c8bcdffda1522c83158981ed92baa045f4
Reviewed-on: https://review.mlplatform.org/364
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/core/CL/kernels/CLReverseKernel.cpp')
| -rw-r--r-- | src/core/CL/kernels/CLReverseKernel.cpp | 15 |
1 files changed, 14 insertions, 1 deletions
diff --git a/src/core/CL/kernels/CLReverseKernel.cpp b/src/core/CL/kernels/CLReverseKernel.cpp index 2859a51ce1..adbdb11c5f 100644 --- a/src/core/CL/kernels/CLReverseKernel.cpp +++ b/src/core/CL/kernels/CLReverseKernel.cpp @@ -80,7 +80,20 @@ void CLReverseKernel::configure(const ICLTensor *input, ICLTensor *output, const // Set kernel build options CLBuildOptions build_opts; build_opts.add_option("-DNUM_REVERSE_DIMS=" + support::cpp11::to_string(axis->info()->dimension(0))); - build_opts.add_option("-DDATA_TYPE=" + get_cl_type_from_data_type(input->info()->data_type())); + switch(input->info()->element_size()) + { + case 1: + build_opts.add_option("-DDATA_TYPE=uchar"); + break; + case 2: + build_opts.add_option("-DDATA_TYPE=ushort"); + break; + case 4: + build_opts.add_option("-DDATA_TYPE=uint"); + break; + default: + ARM_COMPUTE_ERROR("Data type not supported"); + } // Create kernel _kernel = static_cast<cl::Kernel>(CLKernelLibrary::get().create_kernel("reverse", build_opts.options())); |