1 files changed, 6 insertions, 1 deletions
diff --git a/arm_compute/runtime/NEON/functions/NEGEMMLowp.h b/arm_compute/runtime/NEON/functions/NEGEMMLowp.h
index bfb1a494b8..0b0a7742f6 100644
--- a/arm_compute/runtime/NEON/functions/NEGEMMLowp.h
+++ b/arm_compute/runtime/NEON/functions/NEGEMMLowp.h
@@ -32,6 +32,10 @@
 #include "arm_compute/core/NEON/kernels/NEGEMMInterleave4x4Kernel.h"
 #include "arm_compute/core/NEON/kernels/NEGEMMLowpMatrixMultiplyKernel.h"
 #include "arm_compute/core/NEON/kernels/NEGEMMTranspose1xWKernel.h"
+#include "arm_compute/runtime/IMemoryManager.h"
+#include "arm_compute/runtime/MemoryGroup.h"
+
+#include <memory>
 
 namespace arm_compute
 {
@@ -48,7 +52,7 @@ class NEGEMMLowp : public IFunction
 {
 public:
     /** Constructor */
-    NEGEMMLowp();
+    NEGEMMLowp(std::shared_ptr<IMemoryManager> memory_manager = nullptr);
     /** Initialise the kernel's inputs, output
     *
     * @note GEMM_LOWP:  low precision GEMM kernel
@@ -75,6 +79,7 @@ public:
     void run() override;
 
 private:
+    MemoryGroup                    _memory_group;
     NEGEMMInterleave4x4Kernel      _interleave_kernel;
     NEGEMMTranspose1xWKernel       _transpose_kernel;
     NEGEMMLowpMatrixMultiplyKernel _mm_kernel;