diff options
| author | Tim Hall <tim.hall@arm.com> | 2021-05-17 22:57:46 +0100 |
|---|---|---|
| committer | tim.hall <tim.hall@arm.com> | 2021-05-21 15:42:06 +0000 |
| commit | 64556f32ff7bfca6036a6598034464b13b64a4ef (patch) | |
| tree | edf438352185ce88beddbd7f8a2ac62f459aacaa /ethosu/vela/greedy_allocation.py | |
| parent | 0fe2209156719ff0c17fab604dee3fef7d8b8fd7 (diff) | |
| download | ethos-u-vela-64556f32ff7bfca6036a6598034464b13b64a4ef.tar.gz | |
MLBEDSW-4219: Add tensor allocation info to summary
- Moved new tensor allocation info under --verbose-allocation flag
- Tidied up and added histogram to --verbose--allocation print
Signed-off-by: Tim Hall <tim.hall@arm.com>
Change-Id: I76fb5187319aedf86f599f57b766220cafc17326
Diffstat (limited to 'ethosu/vela/greedy_allocation.py')
| -rw-r--r-- | ethosu/vela/greedy_allocation.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/ethosu/vela/greedy_allocation.py b/ethosu/vela/greedy_allocation.py index c68a507d..6f4f8019 100644 --- a/ethosu/vela/greedy_allocation.py +++ b/ethosu/vela/greedy_allocation.py @@ -58,7 +58,7 @@ class GreedyAllocator: def dealloc(self, lr_to_dealloc): self.current_allocs = [(start_addr, lr) for start_addr, lr in self.current_allocs if lr != lr_to_dealloc] - def allocate_live_ranges(self, verbose_allocation, alignment): + def allocate_live_ranges(self, alignment): lrs = set() for lr in self.live_ranges.lrs: lrs.add((lr.start_time, -lr.end_time, lr)) @@ -75,6 +75,6 @@ class GreedyAllocator: return self.memory_required -def allocate_live_ranges(nng, arch, live_ranges, mem_area, alignment, verbose_allocation=False): +def allocate_live_ranges(nng, arch, live_ranges, mem_area, alignment): g = GreedyAllocator(nng, arch, live_ranges, mem_area) - return g.allocate_live_ranges(verbose_allocation, alignment) + return g.allocate_live_ranges(alignment) |