diff options
| author | Tim Hall <tim.hall@arm.com> | 2021-05-27 18:49:40 +0100 |
|---|---|---|
| committer | Tim Hall <tim.hall@arm.com> | 2021-05-27 18:57:39 +0100 |
| commit | d8339a75c9b655c0507e34238078fdad068b4023 (patch) | |
| tree | 36a14726b30760169a83c0356803b480992fade8 /ethosu/vela/nn_graph.py | |
| parent | 64556f32ff7bfca6036a6598034464b13b64a4ef (diff) | |
| download | ethos-u-vela-d8339a75c9b655c0507e34238078fdad068b4023.tar.gz | |
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
Diffstat (limited to 'ethosu/vela/nn_graph.py')
| -rw-r--r-- | ethosu/vela/nn_graph.py | 6 |
1 files changed, 5 insertions, 1 deletions
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 "<nng.CascadedPass strategy=%s x %s '%s', passes=%s, block_configs=%s>" % ( @@ -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 = {} |