commit 8140e1e155d3430992fa46e04ef8938ff09ffd2d
author Anthony Barbier <Anthony.barbier@arm.com> Thu Dec 14 23:48:46 2017 +0000
committer Anthony Barbier <anthony.barbier@arm.com> Wed Jan 24 10:01:21 2018 +0000
tree 9bcf86d01635bfc73e8debd1bda75e6f75b8b406
parent 8a3da6f91f90c566b844d568f4ec43b946915af8

arm_compute v17.12

src/runtime/CL/functions/CLGEMM.cpp

diff --git a/src/runtime/CL/functions/CLGEMM.cpp b/src/runtime/CL/functions/CLGEMM.cpp
index a81d113..ca0228f 100644
--- a/src/runtime/CL/functions/CLGEMM.cpp
+++ b/src/runtime/CL/functions/CLGEMM.cpp
@@ -65,6 +65,9 @@
     const ICLTensor *matrix_a = a;
     const ICLTensor *matrix_b = b;
 
+    // Set the target for the matrix multiply kernel
+    _mm_kernel.set_target(CLScheduler::get().target());
+
     if(_is_interleaved_transposed)
     {
         matrix_a = &_tmp_a;
@@ -95,9 +98,6 @@
 
         // Configure transpose kernel
         _transpose_kernel.configure(b, &_tmp_b);
-
-        // Configure matrix multiply kernel
-        _mm_kernel.set_target(CLScheduler::get().target());
     }
 
     _mm_kernel.configure(matrix_a, matrix_b, output, alpha, _is_interleaved_transposed);