Compute Library
 21.02
OffsetLifetimeManager.cpp
Go to the documentation of this file.
1 /*
2  * Copyright (c) 2017-2020 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"
30 
31 #include <algorithm>
32 #include <cmath>
33 #include <map>
34 #include <vector>
35 
36 namespace arm_compute
37 {
38 namespace
39 {
40 size_t align_offset(size_t offset, size_t alignment)
41 {
42  const size_t remainder = (alignment != 0U) ? offset % alignment : 0U;
43  return (remainder != 0U) ? offset + (alignment - remainder) : offset;
44 }
45 } // namespace
47  : _blob(0)
48 {
49 }
50 
52 {
53  return _blob;
54 }
55 
57 {
58  ARM_COMPUTE_ERROR_ON(allocator == nullptr);
59  return std::make_unique<OffsetMemoryPool>(allocator, _blob);
60 }
61 
63 {
64  return MappingType::OFFSETS;
65 }
66 
67 void OffsetLifetimeManager::update_blobs_and_mappings()
68 {
70  ARM_COMPUTE_ERROR_ON(_active_group == nullptr);
71 
72  // Update blob size
73  size_t max_aggregated_size = 0;
74  std::for_each(std::begin(_free_blobs), std::end(_free_blobs), [&](const Blob & b)
75  {
76  max_aggregated_size += b.max_size;
77  _blob.alignment = std::max(_blob.alignment, b.max_alignment);
78  });
79  max_aggregated_size += _free_blobs.size() * _blob.alignment;
80  _blob.owners = std::max(_blob.owners, _free_blobs.size());
81  _blob.size = std::max(_blob.size, max_aggregated_size);
82 
83  // Calculate group mappings
84  auto &group_mappings = _active_group->mappings();
85  size_t offset = 0;
86  for(auto &free_blob : _free_blobs)
87  {
88  for(auto &bound_element_id : free_blob.bound_elements)
89  {
90  ARM_COMPUTE_ERROR_ON(_active_elements.find(bound_element_id) == std::end(_active_elements));
91  Element &bound_element = _active_elements[bound_element_id];
92  group_mappings[bound_element.handle] = offset;
93  }
94  offset += free_blob.max_size;
95  offset = align_offset(offset, _blob.alignment);
96  ARM_COMPUTE_ERROR_ON(offset > _blob.size);
97  }
98 }
99 } // namespace arm_compute
__global uchar * offset(const Image *img, int x, int y)
Get the pointer position of a Image.
Definition: helpers.h:846
Meta-data information for each blob.
Definition: Types.h:51
SimpleTensor< float > b
Definition: DFT.cpp:157
bool are_all_finalized() const override
Checks if the lifetime of the registered object is complete.
#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
Copyright (c) 2017-2021 Arm Limited.
size_t alignment
Blob alignment.
Definition: Types.h:58
MappingType mapping_type() const override
Returns the type of mappings that the lifetime manager returns.
MappingType
Mapping type.
Definition: Types.h:34
std::unique_ptr< IMemoryPool > create_pool(IAllocator *allocator) override
Creates a memory pool depending on the memory requirements.
void end(TokenStream &in, bool &valid)
Definition: MLGOParser.cpp:290
const info_type & info() const
Accessor to the pool internal configuration meta-data.
void for_each(F &&)
Base case of for_each.
Definition: Utility.h:108
input allocator() -> allocate()
size_t owners
Number of owners in parallel of the blob.
Definition: Types.h:59
Mappings are in offset granularity in the same blob.
Allocator interface.
Definition: IAllocator.h:35
size_t size
Blob size.
Definition: Types.h:57