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>

2 files changed, 23 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
diff --git a/arm_compute/runtime/NEON/functions/assembly/NEGEMMInterleavedWrapper.h b/arm_compute/runtime/NEON/functions/assembly/NEGEMMInterleavedWrapper.h
index cead71ed67..4c5a8d4e59 100644
--- a/arm_compute/runtime/NEON/functions/assembly/NEGEMMInterleavedWrapper.h
+++ b/arm_compute/runtime/NEON/functions/assembly/NEGEMMInterleavedWrapper.h
@@ -93,6 +93,7 @@ private:
     std::vector<PrepareBWorkload>                           _b_workloads{};
     std::vector<MatrixMultiplyWorkload>                     _mm_workloads{};
     std::vector<IScheduler::Workload>                       _workloads{};
+    std::string                                             _tag{};
 };
 
 } // namespace arm_compute