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/live_range.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/live_range.py')
-rw-r--r-- | ethosu/vela/live_range.py | 10 |
1 files changed, 7 insertions, 3 deletions
diff --git a/ethosu/vela/live_range.py b/ethosu/vela/live_range.py index d64f68e0..9f94dd63 100644 --- a/ethosu/vela/live_range.py +++ b/ethosu/vela/live_range.py @@ -27,6 +27,7 @@ from .tensor import MemArea from .tensor import MemType from .tensor import Tensor from .tensor import TensorPurpose +from .utils import progress_print class LiveRange: @@ -231,6 +232,7 @@ def extract_live_ranges_from_cascaded_passes( target_mem_type_set, lr_graph=None, cpu_tensor_alignment=Tensor.AllocationQuantum, + verbose_progress: bool = False, ): if lr_graph is None: lr_graph = LiveRangeGraph() @@ -239,7 +241,8 @@ def extract_live_ranges_from_cascaded_passes( # if subgraph has been processed already, return the lr_graph as is return lr_graph - for cps in sg.cascaded_passes: + for index, cps in enumerate(sg.cascaded_passes): + progress_print(verbose_progress, "Processing cascaded pass", index, sg.cascaded_passes) cps.time = lr_graph.current_time time_for_pass = cps.time @@ -320,9 +323,10 @@ def create_linear_live_range_graph(sg, target_mem_area, target_mem_type_set, lr_ return lr_graph -def extract_live_ranges_from_schedule(sg, target_mem_area, target_mem_type_set, lr_graph): +def extract_live_ranges_from_schedule(sg, target_mem_area, target_mem_type_set, lr_graph, verbose_progress=False): time_for_cascade = {} - for sched_op in sg.sched_ops: + for index, sched_op in enumerate(sg.sched_ops): + progress_print(verbose_progress, "Processing SchedulerOp", index, sg.sched_ops) op_info = sg.schedule.cost_map[sched_op] cascade = op_info.cascade cascade_info = sg.schedule.cascades.get(cascade, None) |