From cc34d5d33938b6cbd9f8b11883f12e8c3097b613 Mon Sep 17 00:00:00 2001 From: Louis Verhaard Date: Thu, 19 Aug 2021 15:15:36 +0200 Subject: MLBEDSW-4976: index errors in scheduler - Fixed index error in memory_snapshot - When removing a cascade, also references are removed Change-Id: I2b35dc52671d8ce115eb32bfdd93584391d1fc6d Signed-off-by: Louis Verhaard --- ethosu/vela/live_range.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) (limited to 'ethosu/vela/live_range.py') diff --git a/ethosu/vela/live_range.py b/ethosu/vela/live_range.py index 7ff1b28d..0b94b197 100644 --- a/ethosu/vela/live_range.py +++ b/ethosu/vela/live_range.py @@ -138,15 +138,11 @@ class LiveRangeGraph: return live_range def update_endtime(self): - self.end_time = 0 - for rng in self.ranges.values(): - self.end_time = max(self.end_time, rng.end_time) + self.end_time = self.current_time return self.end_time + 1 def get_temporal_memory_usage(self, target_mem_area): - if not self.end_time: - self.update_endtime() - usage = np.zeros(self.end_time, dtype=np.int32) + usage = np.zeros(self.update_endtime(), dtype=np.int32) for rng in self.ranges.values(): if rng.mem_area == target_mem_area: # End time is inclusive -- cgit v1.2.1