From 89a8cdd5425521f68674ac23a78790f0f6dc98ed Mon Sep 17 00:00:00 2001 From: wilisa01 Date: Mon, 22 Aug 2022 16:13:06 +0000 Subject: 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 --- ethosu/vela/vela.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'ethosu/vela/vela.py') 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) -- cgit v1.2.1