diff options
| author | Georgios Pinitas <georgios.pinitas@arm.com> | 2019-04-25 18:25:06 +0100 |
|---|---|---|
| committer | Georgios Pinitas <georgios.pinitas@arm.com> | 2019-04-26 15:15:17 +0000 |
| commit | 098516b74a0948707119c64c641a0a8fdf529454 (patch) | |
| tree | db6766ad5e11d33597bf0a1f545397dbbe768558 /arm_compute | |
| parent | cd78347b65cf3a4f4f516a9abb42b8b61f26fd03 (diff) | |
| download | ComputeLibrary-098516b74a0948707119c64c641a0a8fdf529454.tar.gz | |
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 <georgios.pinitas@arm.com>
Reviewed-on: https://review.mlplatform.org/c/1031
Reviewed-by: Giuseppe Rossini <giuseppe.rossini@arm.com>
Reviewed-by: Gian Marco Iodice <gianmarco.iodice@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'arm_compute')
| -rw-r--r-- | arm_compute/runtime/CL/functions/CLFFTConvolutionLayer.h | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/arm_compute/runtime/CL/functions/CLFFTConvolutionLayer.h b/arm_compute/runtime/CL/functions/CLFFTConvolutionLayer.h index 0fd2cf3cb1..2fabe3454a 100644 --- a/arm_compute/runtime/CL/functions/CLFFTConvolutionLayer.h +++ b/arm_compute/runtime/CL/functions/CLFFTConvolutionLayer.h @@ -118,7 +118,7 @@ private: CLPadLayer _pad_input_func; CLPadLayer _pad_weights_func; CLFFT2D _transform_input_func; - CLFFT2D _transform_weights_func; + std::unique_ptr<CLFFT2D> _transform_weights_func; CLFFT2D _itransform_output_func; CLComplexPixelWiseMultiplication _prod_func; CLReductionOperation _reduce_func; |