diff options
| author | Michele Di Giorgio <michele.digiorgio@arm.com> | 2018-04-10 14:24:35 +0100 |
|---|---|---|
| committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:49:37 +0000 |
| commit | fc1d1e2200f3056572b158b8208bac456f48339f (patch) | |
| tree | 754c5ea55f2170afc2503d2fd67759c538660715 /src/core/GLES_COMPUTE/kernels/GCIm2ColKernel.cpp | |
| parent | a72300a5e4d44cdadfe37f69e21f9bf628d19bb3 (diff) | |
| download | ComputeLibrary-fc1d1e2200f3056572b158b8208bac456f48339f.tar.gz | |
COMPMID-959: Add FP32 support to GLES GEMMConvolution
The following kernels were supposed to have FP32 support but this was not the
case because of bugs and missing shaders:
- GCCol2Im
- GCIm2Col
- GCWeightsReshape
Change-Id: Ie6ea464db0612757c71c3d40874e7bb0d60f170a
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/127572
Tested-by: Jenkins <bsgcomp@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Diffstat (limited to 'src/core/GLES_COMPUTE/kernels/GCIm2ColKernel.cpp')
| -rw-r--r-- | src/core/GLES_COMPUTE/kernels/GCIm2ColKernel.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/core/GLES_COMPUTE/kernels/GCIm2ColKernel.cpp b/src/core/GLES_COMPUTE/kernels/GCIm2ColKernel.cpp index eb790471fb..6c896168ed 100644 --- a/src/core/GLES_COMPUTE/kernels/GCIm2ColKernel.cpp +++ b/src/core/GLES_COMPUTE/kernels/GCIm2ColKernel.cpp @@ -113,7 +113,7 @@ void GCIm2ColKernel::configure(const IGCTensor *input, IGCTensor *output, const _convolved_dims = scaled_dimensions(input->info()->dimension(0), input->info()->dimension(1), kernel_dims.width, kernel_dims.height, conv_info, dilation); - _num_elems_processed_per_iteration = 2; + _num_elems_processed_per_iteration = (input->info()->data_type() == DataType::F32) ? 1 : 2; build_opts.emplace("#define KERNEL_WIDTH " + support::cpp11::to_string(kernel_dims.width)); build_opts.emplace("#define KERNEL_HEIGHT " + support::cpp11::to_string(kernel_dims.height)); |