From ac314c25f41e3b2be2ef9073377079584fc88861 Mon Sep 17 00:00:00 2001 From: Anthony Barbier Date: Tue, 11 Sep 2018 17:49:10 +0100 Subject: COMPMID-1563: Fix name of NEGEMMInterleavedWrapper Change-Id: I5f868091cae7bd86eeeb7216d44f32c190c5a604 Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/147804 Tested-by: bsgcomp Reviewed-by: Georgios Pinitas --- arm_compute/core/NEON/kernels/assembly/Helpers.h | 22 ++++++++++++++++++++++ 1 file changed, 22 insertions(+) (limited to 'arm_compute/core') diff --git a/arm_compute/core/NEON/kernels/assembly/Helpers.h b/arm_compute/core/NEON/kernels/assembly/Helpers.h index 0dcba88a95..11c4c08086 100644 --- a/arm_compute/core/NEON/kernels/assembly/Helpers.h +++ b/arm_compute/core/NEON/kernels/assembly/Helpers.h @@ -38,6 +38,28 @@ struct BlockSizes unsigned int strategy_out_height{ 0 }; /**< Number of rows (M) processed by the selected strategy */ }; +/** Calculate the recommended block sizes to use based on the CPU cache sizes and data type + * + * @param[in] ci CPU information + * @param[in] M M dimension. + * @param[in] N N dimension. + * @param[in] K K dimension. + * @param[in] input_type Input data type + * @param[in] use_dot (Optional) If data_type is QASYMM8/U8/S8, then use the dot product instruction ? + * + * @return Recommeded block sizes to use for the given M, N, K dimensions. + */ +BlockSizes calculate_block_sizes_from_data_type(const CPUInfo &ci, unsigned int M, unsigned int N, unsigned int K, DataType input_type, bool use_dot = false); + +/** Get the name of the GEMM strategy which will be used for a given input type + * + * @param[in] input_type Input data type + * @param[in] use_dot (Optional) If data_type is QASYMM8/U8/S8, then use the dot product instruction ? + * + * @return The name of the strategy that will be used + */ +const char *get_strategy_name(DataType input_type, bool use_dot = false); + /** Calculate the recommended block sizes to use based on the CPU cache sizes and the strategy which will be used * * @param[in] ci CPU information -- cgit v1.2.1