diff options
author | Tim Hall <tim.hall@arm.com> | 2022-03-03 17:50:52 +0000 |
---|---|---|
committer | Tim Hall <tim.hall@arm.com> | 2022-05-17 20:06:24 +0100 |
commit | c1be0873d9e28a21c7873793da896e6dd576292f (patch) | |
tree | 8d8392a7ae2a821e6dbc9f048343f9ccbe9e5828 /ethosu/vela/vela.py | |
parent | 3dae1b6088a469f1073222bf249d17c8cdf39dbf (diff) | |
download | ethos-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.py | 2 |
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, |