diff options
| author | Tim Hall <tim.hall@arm.com> | 2020-11-01 21:27:19 +0000 |
|---|---|---|
| committer | Tim Hall <tim.hall@arm.com> | 2020-11-20 12:55:47 +0000 |
| commit | b9b515ca996e1ec5835a2c35033cc6f818f71f12 (patch) | |
| tree | 25d9810755f8fff896cc1d523d0fad90651bc399 /ethosu/vela/live_range.py | |
| parent | 1bd531dec0b4eb745fb8856d14c1aba2b8a73026 (diff) | |
| download | ethos-u-vela-b9b515ca996e1ec5835a2c35033cc6f818f71f12.tar.gz | |
vela: Remove and change CLI options
- Removed unused --show-minimum-possible-allocation
- Change --allocation-alignment to --cpu-tensor-alignment
Signed-off-by: Tim Hall <tim.hall@arm.com>
Change-Id: I00e367c3190aeea08a3f136332711e9accc85ba3
Diffstat (limited to 'ethosu/vela/live_range.py')
| -rw-r--r-- | ethosu/vela/live_range.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/ethosu/vela/live_range.py b/ethosu/vela/live_range.py index a29cafe0..dbc0ce40 100644 --- a/ethosu/vela/live_range.py +++ b/ethosu/vela/live_range.py @@ -236,7 +236,7 @@ def extract_live_ranges_from_cascaded_passes( target_mem_type_set, ignore_subgraph_input_output_tensors=False, lr_graph=None, - allocation_alignment=Tensor.AllocationQuantum, + cpu_tensor_alignment=Tensor.AllocationQuantum, ): if lr_graph is None: lr_graph = LiveRangeGraph() @@ -261,7 +261,7 @@ def extract_live_ranges_from_cascaded_passes( for tens in cps.inputs: if tensor_should_be_ignored(lr_graph, tens, target_mem_area, target_mem_type_set): continue - rng = lr_graph.get_or_create_range(tens, allocation_alignment) + rng = lr_graph.get_or_create_range(tens, cpu_tensor_alignment) rng.mark_usage(time_for_pass) cps_primary_op = cps.passes[0].primary_op @@ -285,7 +285,7 @@ def extract_live_ranges_from_cascaded_passes( for tens in cps.intermediates + cps.outputs: if tensor_should_be_ignored(lr_graph, tens, target_mem_area, target_mem_type_set): continue - rng = lr_graph.get_or_create_range(tens, allocation_alignment) + rng = lr_graph.get_or_create_range(tens, cpu_tensor_alignment) rng.mark_usage(time_for_pass) lr_graph.current_time += 2 @@ -298,7 +298,7 @@ def extract_live_ranges_from_cascaded_passes( for tens in sg.output_tensors: if tensor_should_be_ignored(lr_graph, tens, target_mem_area, target_mem_type_set): continue - rng = lr_graph.get_or_create_range(tens, allocation_alignment) + rng = lr_graph.get_or_create_range(tens, cpu_tensor_alignment) rng.mark_usage(end_time) # Add subgraph to set of processed subgraphs |