commit 186fe683da63dea2dac06e46a412e354d33cd9c2
author Gian Marco Iodice <gianmarco.iodice@arm.com> Tue May 11 11:13:20 2021 +0100
committer Georgios Pinitas <georgios.pinitas@arm.com> Fri May 14 09:34:57 2021 +0000
tree e085313b7cfbb98f1446f74dd8cef937726b02b4
parent 4f7693d8757cf12c33f049c61c63bc689379ab84

Fix the OpenCL convolution layer heuristic for int8

- Call GEMM-based convolution rather than Direct convolution for
  quantized data types

Resolves COMPMID-4497

Change-Id: Idde5808b8a9de69c5196c56538520c0e3d52ba76
Signed-off-by: Gian Marco Iodice <gianmarco.iodice@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/5618
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>

src/runtime/CL/functions/CLConvolutionLayer.cpp

diff --git a/src/runtime/CL/functions/CLConvolutionLayer.cpp b/src/runtime/CL/functions/CLConvolutionLayer.cpp
index 1082a47..96d7cc7 100644
--- a/src/runtime/CL/functions/CLConvolutionLayer.cpp
+++ b/src/runtime/CL/functions/CLConvolutionLayer.cpp
@@ -232,7 +232,7 @@
                     return ConvolutionMethod::FFT;
                 }
             }
-            else
+            else if(is_data_type_float(input->data_type()))
             {
                 if((weights->dimension(idx_h) >= 5) && (input->dimension(idx_c) >= output->dimension(idx_c)) && (CLDirectConvolutionLayer::validate(input, weights, nullptr, output, conv_info, act_info)))
                 {