diff options
author | Francesco.Petrogalli@arm.com <francesco.petrogalli@arm.com> | 2022-03-31 17:55:35 +0000 |
---|---|---|
committer | Francesco Petrogalli <francesco.petrogalli@arm.com> | 2022-04-06 13:41:41 +0000 |
commit | e33c55640b5e6e6af193a41a3376e2a01a321a35 (patch) | |
tree | 3a8282410d5f8e7f57276cbca9e2dfa6d14115d7 /src/cpu/kernels | |
parent | 4c17ba951b76e97102f101a88edbf012b722c732 (diff) | |
download | ComputeLibrary-e33c55640b5e6e6af193a41a3376e2a01a321a35.tar.gz |
[arm_gemm] Use static validate to find arm_gemm kernels.
The static method `CpuGemmAssemblyDispatch::validate` should look into
the list of the available kernels to make sure the one requested by
the user was found.
Formatting changes in the files touched by the patch have been
automatically inserted by the formatting script.
Resolves: ONCPUML-840
Change-Id: Icd650a30e142284a942c64f8a2b72441ee7b3f4e
Signed-off-by: Francesco.Petrogalli@arm.com <francesco.petrogalli@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/7375
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Giorgio Arena <giorgio.arena@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/cpu/kernels')
-rw-r--r-- | src/cpu/kernels/assembly/arm_gemm.hpp | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/src/cpu/kernels/assembly/arm_gemm.hpp b/src/cpu/kernels/assembly/arm_gemm.hpp index e38cc09202..200e04f9a8 100644 --- a/src/cpu/kernels/assembly/arm_gemm.hpp +++ b/src/cpu/kernels/assembly/arm_gemm.hpp @@ -1,5 +1,5 @@ /* - * Copyright (c) 2018-2021 Arm Limited. + * Copyright (c) 2018-2022 Arm Limited. * * SPDX-License-Identifier: MIT * @@ -187,4 +187,7 @@ UniqueGemmCommon<Top, Tret> gemm(const GemmArgs &args, const OutputStage & = {}) template <typename Top, typename Tret, class OutputStage = Nothing> std::vector<KernelDescription> get_compatible_kernels(const GemmArgs &args, const OutputStage & = {}); +template <typename Top, typename Tret, class OutputStage = Nothing> +bool has_opt_gemm(const GemmArgs &args, const OutputStage & = {}); + } // namespace arm_gemm |