commit 26014cf4d0519aef280c8444c60ec34c4e37e3b6
author Georgios Pinitas <georgios.pinitas@arm.com> Mon Sep 09 19:00:57 2019 +0100
committer Georgios Pinitas <georgios.pinitas@arm.com> Tue Sep 17 11:37:02 2019 +0000
tree 7f4b3cb8589c10e95288a722498c9256e8bf5e22
parent caa7deedfe1b0d0020c6099d8f616ec92b1bd5e9

COMPMID-2649: Generalize MemoryGroup.

Avoids any upcasting.

Change-Id: I2181c7c9df59a7fb8a78e11934fbd96058fd39c7
Signed-off-by: Georgios Pinitas <georgios.pinitas@arm.com>
Reviewed-on: https://review.mlplatform.org/c/1918
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Gian Marco Iodice <gianmarco.iodice@arm.com>

arm_compute/runtime/CL/CLTensor.h

diff --git a/arm_compute/runtime/CL/CLTensor.h b/arm_compute/runtime/CL/CLTensor.h
index 65ff4f2..bc72839 100644
--- a/arm_compute/runtime/CL/CLTensor.h
+++ b/arm_compute/runtime/CL/CLTensor.h
@@ -37,7 +37,7 @@
 class ITensorInfo;
 
 /** Basic implementation of the OpenCL tensor interface */
-class CLTensor : public ICLTensor
+class CLTensor : public ICLTensor, public IMemoryManageable
 {
 public:
     /** Constructor */
@@ -68,6 +68,7 @@
     TensorInfo       *info() override;
     const cl::Buffer &cl_buffer() const override;
     CLQuantization    quantization() const override;
+    void associate_memory_group(IMemoryGroup *memory_group) override;
 
 protected:
     // Inherited methods overridden:
@@ -80,5 +81,5 @@
 
 /** OpenCL Image */
 using CLImage = CLTensor;
-}
+} // namespace arm_compute
 #endif /*__ARM_COMPUTE_CLTENSOR_H__ */