diff options
| author | Colm Donelan <colm.donelan@arm.com> | 2022-12-21 15:33:39 +0000 |
|---|---|---|
| committer | Colm Donelan <colm.donelan@arm.com> | 2023-01-05 11:57:23 +0000 |
| commit | 7000d679b5e5ea1df64a25fbe8e97ec23b95aff6 (patch) | |
| tree | e3c5c7ce8dd27d55c4551fcbfe3c0f35110222ee /src/armnn/Runtime.hpp | |
| parent | 0506ef0a099f5ba564af5e110e6857a68f462080 (diff) | |
| download | armnn-7000d679b5e5ea1df64a25fbe8e97ec23b95aff6.tar.gz | |
Deregistering custom allocators in RuntimeImpl destructor.
Signed-off-by: Colm Donelan <colm.donelan@arm.com>
Change-Id: Ib148daf81700d9038d6e91a35c9c17e770c26e84
Diffstat (limited to 'src/armnn/Runtime.hpp')
| -rw-r--r-- | src/armnn/Runtime.hpp | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/src/armnn/Runtime.hpp b/src/armnn/Runtime.hpp index f5dfadf948..9d47b7898d 100644 --- a/src/armnn/Runtime.hpp +++ b/src/armnn/Runtime.hpp @@ -1,5 +1,5 @@ // -// Copyright © 2017 Arm Ltd and Contributors. All rights reserved. +// Copyright © 2017, 2023 Arm Ltd and Contributors. All rights reserved. // SPDX-License-Identifier: MIT // #pragma once @@ -157,6 +157,10 @@ private: /// Profiling Service Instance std::unique_ptr<arm::pipe::IProfilingService> m_ProfilingService; + + /// Keep track of backend ids of the custom allocators that this instance of the runtime added. The + /// destructor can then clean up for this runtime. + std::set<BackendId> m_AllocatorsAddedByThisRuntime; }; } // namespace armnn |