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 <louis.verhaard@arm.com>

1 files changed, 2 insertions, 6 deletions

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