diff options
author | Georgios Pinitas <georgios.pinitas@arm.com> | 2018-05-30 10:17:38 +0100 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:52:54 +0000 |
commit | b95e2106d2e935cb8542f8b9600a2c95ee1c4a5d (patch) | |
tree | ed954a790ad8665f155673792e265fa03556658e /src/runtime/NEON/functions | |
parent | 417d14fe2e5f83a1e17e73d69df2e95bbb1de870 (diff) | |
download | ComputeLibrary-b95e2106d2e935cb8542f8b9600a2c95ee1c4a5d.tar.gz |
COMPMID-1209: Enable memory manager for the GEMM workspace buffer
Change-Id: I125660d412945aa152cb76c78280ca0d52264b86
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/133372
Tested-by: Jenkins <bsgcomp@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Diffstat (limited to 'src/runtime/NEON/functions')
-rw-r--r-- | src/runtime/NEON/functions/NEWinogradConvolutionLayer.cpp | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/src/runtime/NEON/functions/NEWinogradConvolutionLayer.cpp b/src/runtime/NEON/functions/NEWinogradConvolutionLayer.cpp index 8f2c4c4361..1a9c72965b 100644 --- a/src/runtime/NEON/functions/NEWinogradConvolutionLayer.cpp +++ b/src/runtime/NEON/functions/NEWinogradConvolutionLayer.cpp @@ -269,7 +269,9 @@ void NEWinogradConvolutionLayer::configure(const ITensor *input, const ITensor * if(workspace_size > 0) { const unsigned int alignment = 4096; - allocate_workspace(workspace_size, _workspace, &_memory_group, alignment, 1); + // TODO (COMPMID-1248) : Add support for memory manager in NEWinogradConvolutionLayer + // Warning : Do not set a memory group in allocate_workspace, should be done under COMPMID-1248 + allocate_workspace(workspace_size, _workspace, nullptr, alignment, 1); _arm_gemm->set_working_space(reinterpret_cast<float *>(_workspace.buffer())); } |