From a88c5b7563cd639cc9ccb9870fd99b107e7731c4 Mon Sep 17 00:00:00 2001 From: Gian Marco Iodice Date: Tue, 8 Feb 2022 10:58:56 +0000 Subject: Fix performance regression on the first layer of convolution-based model Change-Id: Ia6d7282ee0c91c14dc71505a9f72f7d412d12055 Signed-off-by: Gian Marco Iodice Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/7088 Tested-by: Arm Jenkins Reviewed-by: Giorgio Arena --- src/gpu/cl/operators/ClConv2d.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/gpu/cl/operators/ClConv2d.cpp b/src/gpu/cl/operators/ClConv2d.cpp index 3bb92de843..71f83780c5 100644 --- a/src/gpu/cl/operators/ClConv2d.cpp +++ b/src/gpu/cl/operators/ClConv2d.cpp @@ -271,7 +271,7 @@ ConvolutionMethod ClConv2d::get_convolution_method(const ITensorInfo *src, const // Direct convolution case if(is_direct_valid) { - if((is_large_kernel_sz && workload_gte_8192) || (is_ofm_lte_8 && is_ifm_ge_16)) + if((is_large_kernel_sz && workload_gte_8192 && is_ifm_ge_16) || (is_ofm_lte_8 && is_ifm_ge_16)) { return ConvolutionMethod::DIRECT; } -- cgit v1.2.1