diff options
author | Francesco.Petrogalli@arm.com <francesco.petrogalli@arm.com> | 2022-04-11 16:49:29 +0000 |
---|---|---|
committer | Francesco Petrogalli <francesco.petrogalli@arm.com> | 2022-04-12 14:17:38 +0000 |
commit | 5d606cccaabdfc435734c9fb51e11f14f3724a23 (patch) | |
tree | 1b60ede8bc9ca42eeb3f7368a2a2c8f23afd3589 | |
parent | e33c55640b5e6e6af193a41a3376e2a01a321a35 (diff) | |
download | ComputeLibrary-5d606cccaabdfc435734c9fb51e11f14f3724a23.tar.gz |
Fix CpuGemmAssemblyDispatch::has_opt_impl.
The QASYMM8 case was erroneously using the constructing template
instead of the querying one.
Change-Id: If9257df1aea0aecc3f82235d1cfcbb743fb6b852
Signed-off-by: Francesco.Petrogalli@arm.com <francesco.petrogalli@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/7396
Reviewed-by: Michalis Spyrou <michalis.spyrou@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
-rw-r--r-- | Android.bp | 2 | ||||
-rw-r--r-- | src/cpu/operators/internal/CpuGemmAssemblyDispatch.cpp | 2 |
2 files changed, 2 insertions, 2 deletions
diff --git a/Android.bp b/Android.bp index cf279e0fd9..0d53040e18 100644 --- a/Android.bp +++ b/Android.bp @@ -1,5 +1,5 @@ // -// Copyright © 2020-2021 Arm Ltd. All rights reserved. +// Copyright © 2020-2022 Arm Ltd. All rights reserved. // SPDX-License-Identifier: MIT // diff --git a/src/cpu/operators/internal/CpuGemmAssemblyDispatch.cpp b/src/cpu/operators/internal/CpuGemmAssemblyDispatch.cpp index 496b55ec1c..728333f052 100644 --- a/src/cpu/operators/internal/CpuGemmAssemblyDispatch.cpp +++ b/src/cpu/operators/internal/CpuGemmAssemblyDispatch.cpp @@ -656,7 +656,7 @@ Status CpuGemmAssemblyDispatch::has_opt_impl(const ITensorInfo *a, const ITensor case DataType::QASYMM8: if(d->data_type() == DataType::S32) { - ARM_COMPUTE_RETURN_ERROR_ON_MSG(!(arm_gemm::gemm<uint8_t, uint32_t, arm_gemm::Nothing>(args, {})), + ARM_COMPUTE_RETURN_ERROR_ON_MSG(!(arm_gemm::has_opt_gemm<uint8_t, uint32_t, arm_gemm::Nothing>(args, {})), "We could not find an optimized kernel for U8/QASYMM8 input and S32 output"); } else |