diff options
| author | Raul Farkas <raul.farkas@arm.com> | 2023-04-26 07:49:15 +0100 |
|---|---|---|
| committer | Raul Farkas <raul.farkas@arm.com> | 2023-05-15 11:56:48 +0100 |
| commit | 1c54ac1499da4b1c0de39336c1a9b22e506388b1 (patch) | |
| tree | d3aacca523c9dcf783acdcf7da9ec8921f3c4d05 /ethosu/vela/tensor_allocation.py | |
| parent | fd6f624870b446207b4436cda5bd93dd4ad577ff (diff) | |
| download | ethos-u-vela-1c54ac1499da4b1c0de39336c1a9b22e506388b1.tar.gz | |
MLBEDSW-7390: Add verbose progress option
Add --verbose-progress CLI option used to enable printing progress
information in the compiler driver and scheduler.
Change-Id: I99ac8c6a654e60391d5c11e28b89250405daa53a
Signed-off-by: Raul Farkas <raul.farkas@arm.com>
Diffstat (limited to 'ethosu/vela/tensor_allocation.py')
| -rw-r--r-- | ethosu/vela/tensor_allocation.py | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/ethosu/vela/tensor_allocation.py b/ethosu/vela/tensor_allocation.py index 8c91e2ed..fa9ace17 100644 --- a/ethosu/vela/tensor_allocation.py +++ b/ethosu/vela/tensor_allocation.py @@ -1,4 +1,4 @@ -# SPDX-FileCopyrightText: Copyright 2020-2022 Arm Limited and/or its affiliates <open-source-office@arm.com> +# SPDX-FileCopyrightText: Copyright 2020-2023 Arm Limited and/or its affiliates <open-source-office@arm.com> # # SPDX-License-Identifier: Apache-2.0 # @@ -201,6 +201,7 @@ def allocate( lr_graph=None, cpu_tensor_alignment=Tensor.AllocationQuantum, hillclimb_max_iterations=None, + verbose_progress=False, ): # Allocates addresses to tensors, returns False if tensors could not be fit within max_size lrs = live_range.extract_live_ranges_from_cascaded_passes( @@ -209,6 +210,7 @@ def allocate( mem_type_set, lr_graph=lr_graph, cpu_tensor_alignment=cpu_tensor_alignment, + verbose_progress=verbose_progress, ) total_sz = 0 if lrs.ranges: @@ -235,6 +237,7 @@ def allocate_tensors( mem_type_set, tensor_allocator=TensorAllocator.Greedy, verbose_allocation=False, + verbose_progress=False, lr_graph=None, cpu_tensor_alignment=Tensor.AllocationQuantum, hillclimb_max_iterations=None, @@ -251,6 +254,7 @@ def allocate_tensors( lr_graph=lr_graph, cpu_tensor_alignment=cpu_tensor_alignment, hillclimb_max_iterations=hillclimb_max_iterations, + verbose_progress=verbose_progress, ) if lrs.ranges: |