diff options
| author | wilisa01 <william.isaksson@arm.com> | 2022-08-22 16:13:06 +0000 |
|---|---|---|
| committer | Rickard Bolin <rickard.bolin@arm.com> | 2022-09-01 10:00:50 +0000 |
| commit | 89a8cdd5425521f68674ac23a78790f0f6dc98ed (patch) | |
| tree | 88758bbdff463079820cf91825f6c8116346b05c /ethosu/vela/compiler_driver.py | |
| parent | 8ddd4899892dace88306b3b155dbf47cc47fa4cd (diff) | |
| download | ethos-u-vela-89a8cdd5425521f68674ac23a78790f0f6dc98ed.tar.gz | |
MLBEDSW-6755: Add per-layer performance to CSV file
Dump the current per-layer performance estimation information
that appears on the terminal to a CSV file.
Change-Id: I00e94168704be8c3c674c8779fb807ed28607ccd
Signed-off-by: wilisa01 <william.isaksson@arm.com>
Diffstat (limited to 'ethosu/vela/compiler_driver.py')
| -rw-r--r-- | ethosu/vela/compiler_driver.py | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/ethosu/vela/compiler_driver.py b/ethosu/vela/compiler_driver.py index 1d8756b5..cace0f08 100644 --- a/ethosu/vela/compiler_driver.py +++ b/ethosu/vela/compiler_driver.py @@ -1,4 +1,4 @@ -# Copyright (C) 2020-2021 Arm Limited or its affiliates. All rights reserved. +# Copyright (C) 2020-2022 Arm Limited or its affiliates. All rights reserved. # # SPDX-License-Identifier: Apache-2.0 # @@ -149,7 +149,7 @@ def _check_schedule(nng, arch, scheduler_options): ) -def compiler_driver(nng, arch, options, scheduler_options, network_type): +def compiler_driver(nng, arch, options, scheduler_options, network_type, output_basename): assert verify_graph_health(nng) # Pre-optimisation operator tracking @@ -254,4 +254,6 @@ 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, network_type, options.verbose_performance) + npu_performance.calc_new_performance_for_network( + nng, arch, network_type, options.verbose_performance, output_basename + ) |