COMPMID-417: Fix NEGaussianPyramidOrb.
Change-Id: I9e89136a926f33cc591784d4e7785618bba7b0b2
Reviewed-on: http://mpd-gerrit.cambridge.arm.com/88116
Reviewed-by: Pablo Tello <pablo.tello@arm.com>
Tested-by: Kaizen <jeremy.johnson+kaizengerrit@arm.com>
diff --git a/src/runtime/NEON/functions/NEScale.cpp b/src/runtime/NEON/functions/NEScale.cpp
index 6c5ac3c..0b17e80 100644
--- a/src/runtime/NEON/functions/NEScale.cpp
+++ b/src/runtime/NEON/functions/NEScale.cpp
@@ -86,9 +86,8 @@
}
} // namespace
-NEScale::NEScale(std::shared_ptr<IMemoryManager> memory_manager) // NOLINT
- : _memory_group(std::move(memory_manager)),
- _offsets(),
+NEScale::NEScale() // NOLINT
+ : _offsets(),
_dx(),
_dy(),
_scale_kernel(),
@@ -131,7 +130,6 @@
{
TensorInfo tensor_info_offsets(shape, Format::S32);
_offsets.allocator()->init(tensor_info_offsets);
- _memory_group.manage(&_offsets);
_scale_kernel.configure(input, nullptr, nullptr, &_offsets, output, policy, border_undefined);
@@ -151,11 +149,6 @@
_dx.allocator()->init(tensor_info_dxdy);
_dy.allocator()->init(tensor_info_dxdy);
- // Manage intermediate buffers
- _memory_group.manage(&_offsets);
- _memory_group.manage(&_dx);
- _memory_group.manage(&_dy);
-
_scale_kernel.configure(input, &_dx, &_dy, &_offsets, output, policy, border_undefined);
// Allocate once the configure methods have been called
@@ -181,10 +174,6 @@
void NEScale::run()
{
- _memory_group.acquire();
-
NEScheduler::get().schedule(&_border_handler, Window::DimZ);
NEScheduler::get().schedule(&_scale_kernel, Window::DimY);
-
- _memory_group.release();
}