diff options
-rw-r--r-- | src/runtime/CL/functions/CLGEMM.cpp | 5 |
1 files changed, 5 insertions, 0 deletions
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; } |