diff options
Diffstat (limited to 'src/runtime/CL/functions/CLReduceMean.cpp')
-rw-r--r-- | src/runtime/CL/functions/CLReduceMean.cpp | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/src/runtime/CL/functions/CLReduceMean.cpp b/src/runtime/CL/functions/CLReduceMean.cpp index b2d0f81f50..702ce34a4d 100644 --- a/src/runtime/CL/functions/CLReduceMean.cpp +++ b/src/runtime/CL/functions/CLReduceMean.cpp @@ -140,7 +140,7 @@ Status CLReduceMean::validate(const ITensorInfo *input, const Coordinates &reduc void CLReduceMean::run() { - _memory_group.acquire(); + MemoryGroupResourceScope scope_mg(_memory_group); for(unsigned int i = 0; i < _reduction_ops; ++i) { @@ -151,6 +151,5 @@ void CLReduceMean::run() { _reshape.run(); } - _memory_group.release(); } } // namespace arm_compute |