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/tensor_allocation.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/tensor_allocation.py')
| -rw-r--r-- | ethosu/vela/tensor_allocation.py | 8 |
1 files changed, 1 insertions, 7 deletions
diff --git a/ethosu/vela/tensor_allocation.py b/ethosu/vela/tensor_allocation.py index 4b5e5e42..db261ae5 100644 --- a/ethosu/vela/tensor_allocation.py +++ b/ethosu/vela/tensor_allocation.py @@ -182,14 +182,8 @@ def allocate( cpu_tensor_alignment=Tensor.AllocationQuantum, ): # Allocates addresses to tensors, returns False if tensors could not be fit within max_size - ignore_subgraph_input_output_tensors = False lrs = live_range.extract_live_ranges_from_cascaded_passes( - sg, - mem_area, - mem_type_set, - ignore_subgraph_input_output_tensors=ignore_subgraph_input_output_tensors, - lr_graph=lr_graph, - cpu_tensor_alignment=cpu_tensor_alignment, + sg, mem_area, mem_type_set, lr_graph=lr_graph, cpu_tensor_alignment=cpu_tensor_alignment, ) total_sz = 0 if lrs.ranges: |