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/compiler_driver.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/compiler_driver.py')
| -rw-r--r-- | ethosu/vela/compiler_driver.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/ethosu/vela/compiler_driver.py b/ethosu/vela/compiler_driver.py index 2715c8fe..cec37ef8 100644 --- a/ethosu/vela/compiler_driver.py +++ b/ethosu/vela/compiler_driver.py @@ -60,6 +60,7 @@ class CompilerOptions: verbose_register_command_stream=False, verbose_operators=False, verbose_weights=False, + verbose_performance=False, show_cpu_operations=False, tensor_allocator=TensorAllocator.Greedy, timing=False, @@ -77,6 +78,7 @@ class CompilerOptions: self.verbose_register_command_stream = verbose_register_command_stream self.verbose_operators = verbose_operators self.verbose_weights = verbose_weights + self.verbose_performance = verbose_performance self.show_cpu_operations = show_cpu_operations self.tensor_allocator = tensor_allocator self.timing = timing @@ -250,4 +252,4 @@ def compiler_driver(nng, arch, options, scheduler_options, network_type): cpu_tensor_alignment=options.cpu_tensor_alignment, ) - npu_performance.calc_new_performance_for_network(nng, arch) + npu_performance.calc_new_performance_for_network(nng, arch, network_type, options.verbose_performance) |