From 26014cf4d0519aef280c8444c60ec34c4e37e3b6 Mon Sep 17 00:00:00 2001 From: Georgios Pinitas Date: Mon, 9 Sep 2019 19:00:57 +0100 Subject: COMPMID-2649: Generalize MemoryGroup. Avoids any upcasting. Change-Id: I2181c7c9df59a7fb8a78e11934fbd96058fd39c7 Signed-off-by: Georgios Pinitas Reviewed-on: https://review.mlplatform.org/c/1918 Comments-Addressed: Arm Jenkins Tested-by: Arm Jenkins Reviewed-by: Gian Marco Iodice --- src/graph/backends/CL/CLTensorHandle.cpp | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) (limited to 'src/graph/backends/CL/CLTensorHandle.cpp') diff --git a/src/graph/backends/CL/CLTensorHandle.cpp b/src/graph/backends/CL/CLTensorHandle.cpp index 219d9d0301..891c784cbe 100644 --- a/src/graph/backends/CL/CLTensorHandle.cpp +++ b/src/graph/backends/CL/CLTensorHandle.cpp @@ -1,5 +1,5 @@ /* - * Copyright (c) 2018 ARM Limited. + * Copyright (c) 2018-2019 ARM Limited. * * SPDX-License-Identifier: MIT * @@ -23,8 +23,7 @@ */ #include "arm_compute/graph/backends/CL/CLTensorHandle.h" -#include "arm_compute/core/utils/misc/Cast.h" -#include "arm_compute/runtime/CL/CLMemoryGroup.h" +#include "arm_compute/runtime/IMemoryGroup.h" namespace arm_compute { @@ -52,8 +51,7 @@ void CLTensorHandle::manage(IMemoryGroup *mg) { if(mg != nullptr) { - auto *cl_mg = arm_compute::utils::cast::polymorphic_downcast(mg); - cl_mg->manage(&_tensor); + mg->manage(&_tensor); } } -- cgit v1.2.1