From 8c23ba1c5041467f314a10c1da9147e41d056139 Mon Sep 17 00:00:00 2001 From: SiCong Li Date: Mon, 8 Feb 2021 14:19:23 +0000 Subject: Integrate MLGO into CLGEMM for reshaped kernel Resolves COMPMID-3845 Signed-off-by: SiCong Li Change-Id: I878ea6dc076177095816a75f9bc951326fd095b3 Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/5031 Comments-Addressed: Arm Jenkins Reviewed-by: Gian Marco Iodice Tested-by: Arm Jenkins --- src/runtime/CL/gemm_auto_heuristics/CLGEMMAutoHeuristics.h | 13 +++++++++++++ 1 file changed, 13 insertions(+) (limited to 'src/runtime/CL/gemm_auto_heuristics/CLGEMMAutoHeuristics.h') diff --git a/src/runtime/CL/gemm_auto_heuristics/CLGEMMAutoHeuristics.h b/src/runtime/CL/gemm_auto_heuristics/CLGEMMAutoHeuristics.h index e4fa1a6234..7cb9cab220 100644 --- a/src/runtime/CL/gemm_auto_heuristics/CLGEMMAutoHeuristics.h +++ b/src/runtime/CL/gemm_auto_heuristics/CLGEMMAutoHeuristics.h @@ -82,6 +82,19 @@ GEMMConfigResult select_mlgo_gemm_config_reshaped_only_rhs(const CommonQuery &qu * @return GEMMConfigResult */ GEMMConfigResult select_default_gemm_config_reshaped_only_rhs(const CommonQuery &query); + +/** Select gemm config based on mlgo heuristics + * @param query Query + * @return GEMMConfigResult + */ +GEMMConfigResult select_mlgo_gemm_config_reshaped(const CommonQuery &query); + +/** Select gemm config based on default heuristics + * @param query Query + * @return GEMMConfigResult + */ +GEMMConfigResult select_default_gemm_config_reshaped(const CommonQuery &query); + } // namespace auto_heuristics } // namespace cl_gemm } // namespace arm_compute -- cgit v1.2.1