MLBEDSW-4034: New Scheduler Size or Performance Optimisation
- Merged dev/scheduler at 83639f90e8c828f70de6e29142355a940224959b
Signed-off-by: Tim Hall <tim.hall@arm.com>
Change-Id: I0050529d4b42da93768c7264296434dd877fb5b4
diff --git a/ethosu/vela/nn_graph.py b/ethosu/vela/nn_graph.py
index f810df0..7dc2d72 100644
--- a/ethosu/vela/nn_graph.py
+++ b/ethosu/vela/nn_graph.py
@@ -69,6 +69,7 @@
self.npu_block_type = npu_block_type
self.block_config = None # will be filled in by scheduler
self.shared_buffer = None # will be filled in by scheduler
+ self.scheduling_info = None # will be filled in by scheduler
self.predecessors = []
self.successors = []
@@ -123,6 +124,7 @@
self.predecessors = []
self.successors = []
+ self.sram_used = 0
def __str__(self):
return "<nng.CascadedPass strategy=%s x %s '%s', passes=%s, block_configs=%s>" % (
@@ -149,7 +151,9 @@
self.command_stream_tensor = None
self.flash_tensor = None
# Scratch information locally used in the scheduler
- self.scheduling_info = {}
+ self.schedule = None
+ self.sched_ops = []
+
self.generated_stream_id = None
self.memory_used = {}