diff options
author | Louis Verhaard <louis.verhaard@arm.com> | 2021-03-30 10:18:28 +0200 |
---|---|---|
committer | Louis Verhaard <louis.verhaard@arm.com> | 2021-03-30 16:15:11 +0200 |
commit | 226ecaf4561f421206d1593eac0fa57dd56db82e (patch) | |
tree | 539c69b7f61d0d9452ec1554c861677dfb6a480e /ethosu/vela/greedy_allocation.py | |
parent | 3438c929528583bc019055ad7057c08271b0cee7 (diff) | |
download | ethos-u-vela-226ecaf4561f421206d1593eac0fa57dd56db82e.tar.gz |
Performance improvement in tensor allocation
- Tensor allocation verification was O(N^2), is now closer to O(N)
- Removed a sort in HillClimb allocator
Change-Id: I286a269881490c485cc2b0eeab3b1ecffa8f3df0
Signed-off-by: Louis Verhaard <louis.verhaard@arm.com>
Diffstat (limited to 'ethosu/vela/greedy_allocation.py')
-rw-r--r-- | ethosu/vela/greedy_allocation.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/ethosu/vela/greedy_allocation.py b/ethosu/vela/greedy_allocation.py index 51b07805..c68a507d 100644 --- a/ethosu/vela/greedy_allocation.py +++ b/ethosu/vela/greedy_allocation.py @@ -60,7 +60,7 @@ class GreedyAllocator: def allocate_live_ranges(self, verbose_allocation, alignment): lrs = set() - for lr in self.live_ranges.ranges.values(): + for lr in self.live_ranges.lrs: lrs.add((lr.start_time, -lr.end_time, lr)) lrs = sorted(lrs) |