diff options
| -rw-r--r-- | src/runtime/NEON/functions/NEReductionOperation.cpp | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/src/runtime/NEON/functions/NEReductionOperation.cpp b/src/runtime/NEON/functions/NEReductionOperation.cpp index 853d0ed0ad..fc2e6d2912 100644 --- a/src/runtime/NEON/functions/NEReductionOperation.cpp +++ b/src/runtime/NEON/functions/NEReductionOperation.cpp @@ -178,6 +178,7 @@ void NEReductionOperation::configure(ITensor *input, ITensor *output, unsigned i void NEReductionOperation::run() { + MemoryGroupResourceScope scope_mg(_memory_group); if(_reduction_axis == 0) { NEScheduler::get().schedule(&_fill_border_kernel, Window::DimY); |