aboutsummaryrefslogtreecommitdiff
path: root/src/runtime/NEON/functions/NEWinogradConvolutionLayer.cpp
diff options
context:
space:
mode:
authorGeorgios Pinitas <georgios.pinitas@arm.com>2018-05-30 10:17:38 +0100
committerAnthony Barbier <anthony.barbier@arm.com>2018-11-02 16:52:54 +0000
commitb95e2106d2e935cb8542f8b9600a2c95ee1c4a5d (patch)
treeed954a790ad8665f155673792e265fa03556658e /src/runtime/NEON/functions/NEWinogradConvolutionLayer.cpp
parent417d14fe2e5f83a1e17e73d69df2e95bbb1de870 (diff)
downloadComputeLibrary-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/NEWinogradConvolutionLayer.cpp')
-rw-r--r--src/runtime/NEON/functions/NEWinogradConvolutionLayer.cpp4
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()));
}