From 098516b74a0948707119c64c641a0a8fdf529454 Mon Sep 17 00:00:00 2001 From: Georgios Pinitas Date: Thu, 25 Apr 2019 18:25:06 +0100 Subject: COMPMID-1995: Fix memory management for weights transform in CLFFT Remove memory manager from weights transformation in CLFFTConvolutionLayer. Instead move the transformation function into a unique_ptr and free once the tranformation has been performed. Change-Id: Iabd691b3d517d60fa0127e44eb37d6df873a37d3 Signed-off-by: Georgios Pinitas Reviewed-on: https://review.mlplatform.org/c/1031 Reviewed-by: Giuseppe Rossini Reviewed-by: Gian Marco Iodice Comments-Addressed: Arm Jenkins Tested-by: Arm Jenkins --- src/runtime/CL/functions/CLFFT2D.cpp | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) (limited to 'src/runtime/CL/functions/CLFFT2D.cpp') diff --git a/src/runtime/CL/functions/CLFFT2D.cpp b/src/runtime/CL/functions/CLFFT2D.cpp index 4300fb4e32..165e784786 100644 --- a/src/runtime/CL/functions/CLFFT2D.cpp +++ b/src/runtime/CL/functions/CLFFT2D.cpp @@ -85,11 +85,9 @@ Status CLFFT2D::validate(const ITensorInfo *input, const ITensorInfo *output, co void CLFFT2D::run() { - _memory_group.acquire(); + MemoryGroupResourceScope scope_mg(_memory_group); _first_pass_func.run(); _second_pass_func.run(); - - _memory_group.release(); } } // namespace arm_compute -- cgit v1.2.1