diff options
| author | Fredrik Svedberg <fredrik.svedberg@arm.com> | 2021-10-27 13:58:03 +0200 |
|---|---|---|
| committer | Fredrik Svedberg <fredrik.svedberg@arm.com> | 2021-10-27 13:58:03 +0200 |
| commit | 0ae2848580d376c66d6a3c3b0fb55b30a0234247 (patch) | |
| tree | e36aacb086a50d61a3c413faef3e83f78938ed27 /ethosu/vela/scheduler.py | |
| parent | 2b5939f639d3ceb9fcf75c2edc78d128676119b3 (diff) | |
| download | ethos-u-vela-0ae2848580d376c66d6a3c3b0fb55b30a0234247.tar.gz | |
MLBEDSW-5450 MLCE: Vela to handle skip Tensor
Added checks to avoid merging elementwise op live ranges for subgraph
inputs and outputs, which sometimes caused problems when parts of the
network run on CPU.
Signed-off-by: Fredrik Svedberg <fredrik.svedberg@arm.com>
Change-Id: Id07ab277a205b8550d19a276559f8904b9a4b4be
Diffstat (limited to 'ethosu/vela/scheduler.py')
| -rw-r--r-- | ethosu/vela/scheduler.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/ethosu/vela/scheduler.py b/ethosu/vela/scheduler.py index 2ac47878..782e8d98 100644 --- a/ethosu/vela/scheduler.py +++ b/ethosu/vela/scheduler.py @@ -425,7 +425,7 @@ class Scheduler: lr_graph = live_range.LiveRangeGraph() for mem_area, mem_type_set in memories_list: live_range.extract_live_ranges_from_cascaded_passes( - self.nng.get_root_subgraph(), mem_area, mem_type_set, False, lr_graph, Tensor.AllocationQuantum, + self.nng.get_root_subgraph(), mem_area, mem_type_set, lr_graph, Tensor.AllocationQuantum, ) # Populate time-array with memory used by live ranges @@ -918,7 +918,7 @@ class Scheduler: lr_graph = live_range.LiveRangeGraph() for mem_area, mem_type_set in memories_list: live_range.extract_live_ranges_from_cascaded_passes( - self.nng.get_root_subgraph(), mem_area, mem_type_set, False, lr_graph, Tensor.AllocationQuantum, + self.nng.get_root_subgraph(), mem_area, mem_type_set, lr_graph, Tensor.AllocationQuantum, ) # Iterate over live ranges and evict tensors that doesn't fit |