diff options
Diffstat (limited to 'src/runtime/NEON/functions/NEGEMMConvolutionLayer.cpp')
-rw-r--r-- | src/runtime/NEON/functions/NEGEMMConvolutionLayer.cpp | 16 |
1 files changed, 3 insertions, 13 deletions
diff --git a/src/runtime/NEON/functions/NEGEMMConvolutionLayer.cpp b/src/runtime/NEON/functions/NEGEMMConvolutionLayer.cpp index c405786c80..f63fcb02fd 100644 --- a/src/runtime/NEON/functions/NEGEMMConvolutionLayer.cpp +++ b/src/runtime/NEON/functions/NEGEMMConvolutionLayer.cpp @@ -105,19 +105,9 @@ void NEGEMMConvolutionLayer::prepare() { _impl->weights->mark_as_unused(); } - for(auto &ws : _impl->workspace_tensors) - { - const int slot = ws.slot; - for(auto &m : _impl->aux_mem_req) - { - if(m.slot == slot && m.lifetime == MemoryLifetime::Prepare) - { - auto tensor = ws.tensor.get(); - tensor->allocator()->free(); - break; - } - } - } + + // Release temporary tensors that are only used in prepare stage + release_temporaries<Tensor>(_impl->aux_mem_req, _impl->workspace_tensors); _impl->is_prepared = true; } } |