Compute Library
 21.02
MemoryManagerOnDemand.cpp
Go to the documentation of this file.
1 /*
2  * Copyright (c) 2016-2018 Arm Limited.
3  *
4  * SPDX-License-Identifier: MIT
5  *
6  * Permission is hereby granted, free of charge, to any person obtaining a copy
7  * of this software and associated documentation files (the "Software"), to
8  * deal in the Software without restriction, including without limitation the
9  * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
10  * sell copies of the Software, and to permit persons to whom the Software is
11  * furnished to do so, subject to the following conditions:
12  *
13  * The above copyright notice and this permission notice shall be included in all
14  * copies or substantial portions of the Software.
15  *
16  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
19  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
22  * SOFTWARE.
23  */
25 
26 #include "arm_compute/core/Error.h"
29 
30 #include <memory>
31 
32 namespace arm_compute
33 {
34 MemoryManagerOnDemand::MemoryManagerOnDemand(std::shared_ptr<ILifetimeManager> lifetime_manager, std::shared_ptr<IPoolManager> pool_manager)
35  : _lifetime_mgr(std::move(lifetime_manager)), _pool_mgr(std::move(pool_manager))
36 {
37  ARM_COMPUTE_ERROR_ON_MSG(!_lifetime_mgr, "Lifetime manager not specified correctly!");
38  ARM_COMPUTE_ERROR_ON_MSG(!_pool_mgr, "Pool manager not specified correctly!");
39 }
40 
42 {
43  return _lifetime_mgr.get();
44 }
45 
47 {
48  return _pool_mgr.get();
49 }
50 
52 {
53  ARM_COMPUTE_ERROR_ON(!_lifetime_mgr);
54  ARM_COMPUTE_ERROR_ON(!_pool_mgr);
55  ARM_COMPUTE_ERROR_ON_MSG(!_lifetime_mgr->are_all_finalized(), "All the objects have not been finalized!");
56  ARM_COMPUTE_ERROR_ON_MSG(_pool_mgr->num_pools() != 0, "Pool manager already contains pools!");
57 
58  // Create pools
59  auto pool_template = _lifetime_mgr->create_pool(&allocator);
60  for(int i = num_pools; i > 1; --i)
61  {
62  auto pool = pool_template->duplicate();
63  _pool_mgr->register_pool(std::move(pool));
64  }
65  _pool_mgr->register_pool(std::move(pool_template));
66 }
67 
69 {
70  ARM_COMPUTE_ERROR_ON_MSG(!_pool_mgr, "Pool manager not specified correctly!");
71  _pool_mgr->clear_pools();
72 }
73 } //namespace arm_compute
Memory pool manager interface.
Definition: IPoolManager.h:34
MemoryManagerOnDemand(std::shared_ptr< ILifetimeManager > lifetime_manager, std::shared_ptr< IPoolManager > pool_manager)
Default Constructor.
#define ARM_COMPUTE_ERROR_ON(cond)
If the condition is true then an error message is printed and an exception thrown.
Definition: Error.h:466
IPoolManager * pool_manager() override
Returns the pool manager used by the memory manager.
Copyright (c) 2017-2021 Arm Limited.
Interface for managing the lifetime of objects.
void clear() override
Clears the pool manager.
ILifetimeManager * lifetime_manager() override
Returns the lifetime manager used by the memory manager.
#define ARM_COMPUTE_ERROR_ON_MSG(cond, msg)
Definition: Error.h:456
input allocator() -> allocate()
void populate(IAllocator &allocator, size_t num_pools) override
Populates the pool manager with the given number of pools.
Allocator interface.
Definition: IAllocator.h:35