From de6cb64b156108cf48e34c80241bd70f0307a588 Mon Sep 17 00:00:00 2001 From: "erik.andersson@arm.com" Date: Wed, 2 Feb 2022 14:03:15 +0100 Subject: 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 Change-Id: I34785840c905a79750a62863773015b00fb43387 --- ethosu/vela/live_range.py | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) (limited to 'ethosu/vela/live_range.py') 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 "" % (self.name, self.start_time, self.end_time) + return ( + f"" + ) __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 -- cgit v1.2.1