From 5d606cccaabdfc435734c9fb51e11f14f3724a23 Mon Sep 17 00:00:00 2001 From: "Francesco.Petrogalli@arm.com" Date: Mon, 11 Apr 2022 16:49:29 +0000 Subject: 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 Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/7396 Reviewed-by: Michalis Spyrou Comments-Addressed: Arm Jenkins Tested-by: Arm Jenkins --- Android.bp | 2 +- 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(args, {})), + ARM_COMPUTE_RETURN_ERROR_ON_MSG(!(arm_gemm::has_opt_gemm(args, {})), "We could not find an optimized kernel for U8/QASYMM8 input and S32 output"); } else -- cgit v1.2.1