From 4cdd6b80754b3abbf54650d9359cf940a4aaf772 Mon Sep 17 00:00:00 2001 From: Dana Zlotnik Date: Thu, 7 Oct 2021 15:31:54 +0300 Subject: Implement Minimum Workload Size (MWS) in all CPPKernels used by small networks * create get_mws method in ICPPKernel class that retuns default value for all kernels * overwrite the default value for all the kernels used by small networks (according to banchmark case) Resolves COMPMID-4648 Change-Id: I46d7cae61217213279d2ee740edc73f600b6d576 Signed-off-by: Dana Zlotnik Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/6412 Tested-by: Arm Jenkins Reviewed-by: SiCong Li Comments-Addressed: Arm Jenkins --- src/core/NEON/kernels/NEPadLayerKernel.h | 9 +++++++++ 1 file changed, 9 insertions(+) (limited to 'src/core/NEON/kernels/NEPadLayerKernel.h') diff --git a/src/core/NEON/kernels/NEPadLayerKernel.h b/src/core/NEON/kernels/NEPadLayerKernel.h index 00cda7dc22..b3b0725af8 100644 --- a/src/core/NEON/kernels/NEPadLayerKernel.h +++ b/src/core/NEON/kernels/NEPadLayerKernel.h @@ -79,6 +79,15 @@ public: // Inherited methods overridden: void run(const Window &window, const ThreadInfo &info) override; + /** Return minimum workload size of the relevant kernel + * + * @param[in] platform The CPU platform used to create the context. + * @param[in] thread_count Number of threads in the execution. + * + * @return[out] small_network_mws Minimum workload size for requsted configuration. + */ + size_t get_mws(const CPUInfo &platform, size_t thread_count) const override; + private: /** Template function to run the padding function with constant padding * -- cgit v1.2.1