diff options
Diffstat (limited to 'src/runtime/CL/functions/CLGEMM.cpp')
| -rw-r--r-- | src/runtime/CL/functions/CLGEMM.cpp | 11 |
1 files changed, 9 insertions, 2 deletions
diff --git a/src/runtime/CL/functions/CLGEMM.cpp b/src/runtime/CL/functions/CLGEMM.cpp index 8d460142e5..fc56a0bd6b 100644 --- a/src/runtime/CL/functions/CLGEMM.cpp +++ b/src/runtime/CL/functions/CLGEMM.cpp @@ -1,5 +1,5 @@ /* - * Copyright (c) 2017-2019 ARM Limited. + * Copyright (c) 2017-2020 ARM Limited. * * SPDX-License-Identifier: MIT * @@ -109,7 +109,14 @@ CLGEMM::GEMMType CLGEMM::select_gemm_type(unsigned int m, unsigned int n, unsign { if((m == 1) || (!reshape_b_only_on_first_run)) { - gemm_type = GEMMType::RESHAPED_ONLY_RHS; + if((n > k) && gpu_target_is_in(gpu_target, GPUTarget::G71)) + { + gemm_type = GEMMType::NATIVE; + } + else + { + gemm_type = GEMMType::RESHAPED_ONLY_RHS; + } } else { |