commit 5dc732e95b64c793e3204515df51b403d05e79bb
author Gunes Bayir <gunes.bayir@arm.com> Tue Feb 07 09:46:53 2023 +0000
committer Gunes Bayir <gunes.bayir@arm.com> Thu Feb 09 09:39:15 2023 +0000
tree a6d3cac3bb4bf0825f1b4699091880a913d6a8a2
parent 7d9a626aaba9837cb82d189a9c4f0bcef58825bb

Fix performance regression in Transposed Convolution

Resolves: COMPMID-5849

Change-Id: I86f8bbc1f3a7c12c66d5ad8fcd74dd9e69629aa0
Signed-off-by: Gunes Bayir <gunes.bayir@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/9102
Reviewed-by: Jakub Sujak <jakub.sujak@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Benchmark: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Dynamic-Fusion: Jakub Sujak <jakub.sujak@arm.com>

src/runtime/CL/functions/CLDeconvolutionLayer.cpp

diff --git a/src/runtime/CL/functions/CLDeconvolutionLayer.cpp b/src/runtime/CL/functions/CLDeconvolutionLayer.cpp
index 56e9dae..5c25cba 100644
--- a/src/runtime/CL/functions/CLDeconvolutionLayer.cpp
+++ b/src/runtime/CL/functions/CLDeconvolutionLayer.cpp
@@ -150,10 +150,12 @@
 
     const size_t idx_w = get_data_layout_dimension_index(data_layout, DataLayoutDimension::WIDTH);
     const size_t idx_h = get_data_layout_dimension_index(data_layout, DataLayoutDimension::HEIGHT);
+    const size_t idx_n = get_data_layout_dimension_index(data_layout, DataLayoutDimension::BATCHES);
+    const size_t ofm   = weights->tensor_shape()[idx_n];
 
     if(weights->dimension(idx_w) != deconv_info.stride().first || weights->dimension(idx_h) != deconv_info.stride().second)
     {
-        if(input->data_layout() == DataLayout::NHWC)
+        if(input->data_layout() == DataLayout::NHWC && ofm <= 16)
         {
             return DeconvolutionMethod::DIRECT;
         }