From cda0c38373b2f114509392ba16ef04e8c1e0f819 Mon Sep 17 00:00:00 2001 From: Gian Marco Iodice Date: Mon, 23 Apr 2018 16:16:22 +0100 Subject: COMPMID-1080 - Fix CLFullyConnectedLayer validate() tests failing is_interleaved_transposed was set to true on Midgard when m == 1 Change-Id: I991be804073c12f6be894abbfdd010744559772d Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/128678 Reviewed-by: Georgios Pinitas Tested-by: Jenkins --- src/runtime/CL/functions/CLGEMM.cpp | 5 +++++ 1 file changed, 5 insertions(+) (limited to 'src/runtime/CL/functions/CLGEMM.cpp') diff --git a/src/runtime/CL/functions/CLGEMM.cpp b/src/runtime/CL/functions/CLGEMM.cpp index bff5781300..37fa0c5ba2 100644 --- a/src/runtime/CL/functions/CLGEMM.cpp +++ b/src/runtime/CL/functions/CLGEMM.cpp @@ -57,6 +57,11 @@ inline bool is_interleaved_transposed(int m, int n, int k, DataType data_type, b flag = false; } } + else + { + // We reshape the matrices only if we do not have the vector-by-matrix case and we reshape the matrix B only once + flag = m != 1 && reshape_b_only_on_first_run; + } return flag; } -- cgit v1.2.1