diff options
Diffstat (limited to 'src/runtime')
-rw-r--r-- | src/runtime/CL/functions/CLGEMM.cpp | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/src/runtime/CL/functions/CLGEMM.cpp b/src/runtime/CL/functions/CLGEMM.cpp index 7408054127..07b19421d6 100644 --- a/src/runtime/CL/functions/CLGEMM.cpp +++ b/src/runtime/CL/functions/CLGEMM.cpp @@ -101,6 +101,7 @@ void CLGEMM::configure(const ICLTensor *a, const ICLTensor *b, const ICLTensor * _transpose_kernel.configure(b, &_tmp_b); // Configure matrix multiply kernel + _mm_kernel.set_target(CLScheduler::get().target()); _mm_kernel.configure(&_tmp_a, &_tmp_b, output, alpha); // Allocate intermediate tensors |