diff options
| author | Gian Marco Iodice <gianmarco.iodice@arm.com> | 2017-08-21 09:15:58 +0100 |
|---|---|---|
| committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:35:24 +0000 |
| commit | 75af28e6936f084fdfa65add7631cd2cd2050b0c (patch) | |
| tree | 4c39b5c9aa2dc6d83b35f6da4d87e81734458c5e /src/core/CL/kernels/CLDirectConvolutionLayerKernel.cpp | |
| parent | cc65bbe73dc4e376108fedb82ce06df983536595 (diff) | |
| download | ComputeLibrary-75af28e6936f084fdfa65add7631cd2cd2050b0c.tar.gz | |
COMPMID-417 - Fixed call of direct convolution 1x1 for bifrost
Change-Id: Ic4e56e8881b8c66758e67c486514ec397cf43f8e
Reviewed-on: http://mpd-gerrit.cambridge.arm.com/84592
Reviewed-by: Moritz Pflanzer <moritz.pflanzer@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Tested-by: Kaizen <jeremy.johnson+kaizengerrit@arm.com>
Diffstat (limited to 'src/core/CL/kernels/CLDirectConvolutionLayerKernel.cpp')
| -rw-r--r-- | src/core/CL/kernels/CLDirectConvolutionLayerKernel.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/core/CL/kernels/CLDirectConvolutionLayerKernel.cpp b/src/core/CL/kernels/CLDirectConvolutionLayerKernel.cpp index b012f59bd2..9d1c87d9e1 100644 --- a/src/core/CL/kernels/CLDirectConvolutionLayerKernel.cpp +++ b/src/core/CL/kernels/CLDirectConvolutionLayerKernel.cpp @@ -108,7 +108,7 @@ void CLDirectConvolutionLayerKernel::configure(const ICLTensor *input, const ICL options.emplace("-DHAS_BIAS"); } - if((gpu_target == GPUTarget::BIFROST) && (kernel_size <= 5) && (_conv_stride_x == 1) && (_conv_stride_y == 1) && (input->info()->data_type() == DataType::F32)) + if((gpu_target == GPUTarget::BIFROST) && (kernel_size <= 5) && (kernel_size != 1) && (_conv_stride_x == 1) && (_conv_stride_y == 1) && (input->info()->data_type() == DataType::F32)) { options.emplace("-DWEIGHTS_DEPTH=" + support::cpp11::to_string(_weights->info()->dimension(2))); |