diff options
author | Georgios Pinitas <georgios.pinitas@arm.com> | 2018-04-24 15:14:12 +0100 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:49:54 +0000 |
commit | 82b51482479951cf133c223eb81aae291cb4d590 (patch) | |
tree | f7ce7b0f169755234f788d68e296a769230e9964 /src/runtime/CL/functions/CLWinogradConvolutionLayer.cpp | |
parent | 35aa6a36e7ef9302efd554aac9b84153ad7c6a9e (diff) | |
download | ComputeLibrary-82b51482479951cf133c223eb81aae291cb4d590.tar.gz |
COMPMID-959: Sets memory manager to CLWinograd
-Sets memory manager to Winograd functions
-Marks CLGEMM inputs as unused if needed
Change-Id: I425a3f864c756e0e2b4da895e1730b8822149ba8
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/128891
Tested-by: Jenkins <bsgcomp@arm.com>
Reviewed-by: Gian Marco Iodice <gianmarco.iodice@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Diffstat (limited to 'src/runtime/CL/functions/CLWinogradConvolutionLayer.cpp')
-rw-r--r-- | src/runtime/CL/functions/CLWinogradConvolutionLayer.cpp | 11 |
1 files changed, 9 insertions, 2 deletions
diff --git a/src/runtime/CL/functions/CLWinogradConvolutionLayer.cpp b/src/runtime/CL/functions/CLWinogradConvolutionLayer.cpp index 86ccddac88..65747cf5d7 100644 --- a/src/runtime/CL/functions/CLWinogradConvolutionLayer.cpp +++ b/src/runtime/CL/functions/CLWinogradConvolutionLayer.cpp @@ -151,8 +151,6 @@ void CLWinogradConvolutionLayer::run() { // Run filter transform CLScheduler::get().enqueue(_filter_transform, false); - - _is_first_run = false; } _memory_group.acquire(); @@ -163,6 +161,13 @@ void CLWinogradConvolutionLayer::run() // Run batched matrix multiplication _batched_mm.run(); + // Release reshaped weights if marked unused by CLGEMM + if(_is_first_run && !_input1.is_used()) + { + CLScheduler::get().queue().finish(); + _input1.allocator()->free(); + } + // Run output transform CLScheduler::get().enqueue(_output_transform); @@ -172,4 +177,6 @@ void CLWinogradConvolutionLayer::run() } _memory_group.release(); + + _is_first_run = false; } |