diff options
author | erik.andersson@arm.com <erik.andersson@arm.com> | 2022-02-02 14:03:15 +0100 |
---|---|---|
committer | tim.hall <tim.hall@arm.com> | 2022-02-08 10:00:35 +0000 |
commit | de6cb64b156108cf48e34c80241bd70f0307a588 (patch) | |
tree | 2245c836aa30366eef6908bd543f987547f5c0f5 /ethosu/vela/live_range.py | |
parent | 8d57aaa722a4f0430d7e7520d194d1df5394eb6c (diff) | |
download | ethos-u-vela-de6cb64b156108cf48e34c80241bd70f0307a588.tar.gz |
MLBEDSW-5839: Port of improved spilling behaviour
Ported the improved spilling behaviour from Regor
into Vela. This replaces use_fast_storage_for_feature_maps
with allocate_feature_maps and introduces the class called
FastStorageComponentAllocator.
Signed-off-by: erik.andersson@arm.com <erik.andersson@arm.com>
Change-Id: I34785840c905a79750a62863773015b00fb43387
Diffstat (limited to 'ethosu/vela/live_range.py')
-rw-r--r-- | ethosu/vela/live_range.py | 11 |
1 files changed, 7 insertions, 4 deletions
diff --git a/ethosu/vela/live_range.py b/ethosu/vela/live_range.py index 1aaaadd5..fc94e9dd 100644 --- a/ethosu/vela/live_range.py +++ b/ethosu/vela/live_range.py @@ -42,7 +42,10 @@ class LiveRange: self.add_tensor(tens) def __str__(self): - return "<live_range.LiveRange: '%s' start_time=%s, end_time=%s>" % (self.name, self.start_time, self.end_time) + return ( + f"<live_range.LiveRange: {self.start_time}-{self.end_time}, " + f"size={self.size}, '{self.name}' #:{len(self.tensors)}>" + ) __repr__ = __str__ @@ -142,10 +145,10 @@ class LiveRangeGraph: def get_temporal_memory_usage(self, target_mem_area): usage = np.zeros(self.update_endtime(), dtype=np.int32) - for rng in self.ranges.values(): - if rng.mem_area == target_mem_area: + for lr in self.lrs: + if lr.mem_area == target_mem_area: # End time is inclusive - usage[rng.start_time : rng.end_time + 1] += rng.size + usage[lr.start_time : lr.end_time + 1] += lr.size return usage |