diff options
| author | Giorgio Arena <giorgio.arena@arm.com> | 2021-08-16 12:29:27 +0100 |
|---|---|---|
| committer | Giorgio Arena <giorgio.arena@arm.com> | 2021-08-19 10:46:51 +0000 |
| commit | 73df9310f4e94e43597f283307e3cde0653d8bae (patch) | |
| tree | b0a286dc00496cde6b98c1f96119ae5c830b1815 /src/runtime/IWeightsManager.cpp | |
| parent | df5bcb67e895610c0a7360753817fea13faabd4e (diff) | |
| download | ComputeLibrary-73df9310f4e94e43597f283307e3cde0653d8bae.tar.gz | |
Address comments on avoiding releasing weights if used by multiple functions
Signed-off-by: Giorgio Arena <giorgio.arena@arm.com>
Change-Id: I0b59c5326f5fcbc322fbeb864197ea999de6bd56
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/6112
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Gian Marco Iodice <gianmarco.iodice@arm.com>
Reviewed-by: SiCong Li <sicong.li@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/runtime/IWeightsManager.cpp')
| -rw-r--r-- | src/runtime/IWeightsManager.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/runtime/IWeightsManager.cpp b/src/runtime/IWeightsManager.cpp index 8f27795285..373c50c73d 100644 --- a/src/runtime/IWeightsManager.cpp +++ b/src/runtime/IWeightsManager.cpp @@ -166,7 +166,7 @@ void IWeightsManager::release(const ITensor *weights) } } -void IWeightsManager::mark_as_unused(const ITensor *weights) +void IWeightsManager::pre_mark_as_unused(const ITensor *weights) { if(weights == nullptr || !are_weights_managed(weights)) { |