aboutsummaryrefslogtreecommitdiff
path: root/ethosu/vela/vela.py
diff options
context:
space:
mode:
authorwilisa01 <william.isaksson@arm.com>2022-08-22 16:13:06 +0000
committerRickard Bolin <rickard.bolin@arm.com>2022-09-01 10:00:50 +0000
commit89a8cdd5425521f68674ac23a78790f0f6dc98ed (patch)
tree88758bbdff463079820cf91825f6c8116346b05c /ethosu/vela/vela.py
parent8ddd4899892dace88306b3b155dbf47cc47fa4cd (diff)
downloadethos-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/vela.py')
-rw-r--r--ethosu/vela/vela.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/ethosu/vela/vela.py b/ethosu/vela/vela.py
index 1de437bb..a42b2188 100644
--- a/ethosu/vela/vela.py
+++ b/ethosu/vela/vela.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
#
@@ -75,7 +75,7 @@ def process(input_name, enable_debug_db, arch, model_reader_options, compiler_op
print("Model reading took %f s" % (stop - start))
start = time.time()
- compiler_driver.compiler_driver(nng, arch, compiler_options, scheduler_options, network_type)
+ compiler_driver.compiler_driver(nng, arch, compiler_options, scheduler_options, network_type, output_basename)
summary_csv_file = "{0}_summary_{1}.csv".format(output_basename, arch.system_config)
stats_writer.write_summary_metrics_csv(nng, summary_csv_file, arch)