From d8339a75c9b655c0507e34238078fdad068b4023 Mon Sep 17 00:00:00 2001 From: Tim Hall Date: Thu, 27 May 2021 18:49:40 +0100 Subject: MLBEDSW-4034: New Scheduler Size or Performance Optimisation - Merged dev/scheduler at 83639f90e8c828f70de6e29142355a940224959b Signed-off-by: Tim Hall Change-Id: I0050529d4b42da93768c7264296434dd877fb5b4 --- ethosu/vela/nn_graph.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) (limited to 'ethosu/vela/nn_graph.py') diff --git a/ethosu/vela/nn_graph.py b/ethosu/vela/nn_graph.py index f810df0f..7dc2d722 100644 --- a/ethosu/vela/nn_graph.py +++ b/ethosu/vela/nn_graph.py @@ -69,6 +69,7 @@ class Pass: 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 @@ class CascadedPass: self.predecessors = [] self.successors = [] + self.sram_used = 0 def __str__(self): return "" % ( @@ -149,7 +151,9 @@ class Subgraph: 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 = {} -- cgit v1.2.1