aboutsummaryrefslogtreecommitdiff
path: root/ethosu/vela/vela.py
diff options
context:
space:
mode:
authorTim Hall <tim.hall@arm.com>2022-03-03 17:50:52 +0000
committerTim Hall <tim.hall@arm.com>2022-05-17 20:06:24 +0100
commitc1be0873d9e28a21c7873793da896e6dd576292f (patch)
tree8d8392a7ae2a821e6dbc9f048343f9ccbe9e5828 /ethosu/vela/vela.py
parent3dae1b6088a469f1073222bf249d17c8cdf39dbf (diff)
downloadethos-u-vela-c1be0873d9e28a21c7873793da896e6dd576292f.tar.gz
MLBEDSW-6271: MLCE: Layer wise Utilization info from Vela
- Added support to print per operator sram usage and performance information - Added new CLI option --verbose-performance to control this feature Signed-off-by: Tim Hall <tim.hall@arm.com> Change-Id: I368599b410e5d441d9804871fc51b7a1049d85b3
Diffstat (limited to 'ethosu/vela/vela.py')
-rw-r--r--ethosu/vela/vela.py2
1 files changed, 2 insertions, 0 deletions
diff --git a/ethosu/vela/vela.py b/ethosu/vela/vela.py
index efe8edbb..2108e209 100644
--- a/ethosu/vela/vela.py
+++ b/ethosu/vela/vela.py
@@ -341,6 +341,7 @@ def main(args=None):
)
parser.add_argument("--verbose-operators", action="store_true", help="Verbose operator list")
parser.add_argument("--verbose-weights", action="store_true", help="Verbose weights information")
+ parser.add_argument("--verbose-performance", action="store_true", help="Verbose performance information")
parser.add_argument(
"--show-cpu-operations", action="store_true", help="Show the operations that fall back to the CPU"
)
@@ -476,6 +477,7 @@ def main(args=None):
verbose_register_command_stream=args.verbose_register_command_stream,
verbose_operators=args.verbose_operators,
verbose_weights=args.verbose_weights,
+ verbose_performance=args.verbose_performance,
show_cpu_operations=args.show_cpu_operations,
tensor_allocator=args.tensor_allocator,
timing=args.timing,