diff options
Diffstat (limited to 'src/runtime/NEON/functions/NEGEMM.cpp')
-rw-r--r-- | src/runtime/NEON/functions/NEGEMM.cpp | 8 |
1 files changed, 2 insertions, 6 deletions
diff --git a/src/runtime/NEON/functions/NEGEMM.cpp b/src/runtime/NEON/functions/NEGEMM.cpp index 914f088bf5..55bcc45d12 100644 --- a/src/runtime/NEON/functions/NEGEMM.cpp +++ b/src/runtime/NEON/functions/NEGEMM.cpp @@ -238,16 +238,14 @@ void NEGEMM::run() { prepare(); + MemoryGroupResourceScope scope_mg(_memory_group); + if(_asm_glue.is_configured()) { - _memory_group.acquire(); _asm_glue.run(); - _memory_group.release(); } else { - _memory_group.acquire(); - if(!_run_vector_matrix_multiplication) { // Run interleave kernel @@ -262,8 +260,6 @@ void NEGEMM::run() NEScheduler::get().schedule(&_mm_kernel, _run_vector_matrix_multiplication ? Window::DimX : Window::DimY); - _memory_group.release(); - // Run matrix addition kernel if(_run_addition) { |