diff options
author | Anthony Barbier <anthony.barbier@arm.com> | 2018-09-11 17:49:10 +0100 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:54:54 +0000 |
commit | ac314c25f41e3b2be2ef9073377079584fc88861 (patch) | |
tree | a654217d60a961288f0a0b3fccc7e5714a2f05bc /arm_compute/core/NEON/kernels | |
parent | cbf39c63a6eb89a2c80b2338afc374081803d79d (diff) | |
download | ComputeLibrary-ac314c25f41e3b2be2ef9073377079584fc88861.tar.gz |
COMPMID-1563: Fix name of NEGEMMInterleavedWrapper
Change-Id: I5f868091cae7bd86eeeb7216d44f32c190c5a604
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/147804
Tested-by: bsgcomp <bsgcomp@arm.com>
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Diffstat (limited to 'arm_compute/core/NEON/kernels')
-rw-r--r-- | arm_compute/core/NEON/kernels/assembly/Helpers.h | 22 |
1 files changed, 22 insertions, 0 deletions
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 |