From 1509e4bfcfd4b613e2f1ad584c51b80b5fb05a8c Mon Sep 17 00:00:00 2001 From: Georgios Pinitas Date: Mon, 28 Jan 2019 10:01:50 +0000 Subject: COMPMID-1823: Increase scheduling granularity in NEGemmInterleaved. Change-Id: I68c6744885c8aa56a882cf5267061deeebcdc197 Signed-off-by: Georgios Pinitas Reviewed-on: https://review.mlplatform.org/623 Tested-by: Arm Jenkins Reviewed-by: Pablo Marquez --- src/core/NEON/kernels/assembly/NEGEMMInterleavedStrategies.h | 9 +++++++++ 1 file changed, 9 insertions(+) (limited to 'src/core') diff --git a/src/core/NEON/kernels/assembly/NEGEMMInterleavedStrategies.h b/src/core/NEON/kernels/assembly/NEGEMMInterleavedStrategies.h index da6ef2dea9..26d9e9999d 100644 --- a/src/core/NEON/kernels/assembly/NEGEMMInterleavedStrategies.h +++ b/src/core/NEON/kernels/assembly/NEGEMMInterleavedStrategies.h @@ -52,6 +52,11 @@ class IInterleavedStrategy public: /** Virtual Destructor */ virtual ~IInterleavedStrategy() = default; + /** Return output height of the interleaved strategy + * + * @return Output height of strategy + */ + virtual unsigned int out_height() const = 0; /** Instantiate and configure a prepareB Kernel * * @param[in] b Input tensor B. @@ -117,6 +122,10 @@ public: public: // Inherited methods overridden + unsigned int out_height() const override + { + return strategy::out_height(); + } std::unique_ptr instantiate_prepareB(const ITensor *b, ITensor *transformed_b, const INEGEMMWrapperKernel::Params ¶ms, -- cgit v1.2.1