commit 89a8cdd5425521f68674ac23a78790f0f6dc98ed
author wilisa01 <william.isaksson@arm.com> Mon Aug 22 16:13:06 2022 +0000
committer Rickard Bolin <rickard.bolin@arm.com> Thu Sep 01 10:00:50 2022 +0000
tree 88758bbdff463079820cf91825f6c8116346b05c
parent 8ddd4899892dace88306b3b155dbf47cc47fa4cd

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>

ethosu/vela/compiler_driver.py

diff --git a/ethosu/vela/compiler_driver.py b/ethosu/vela/compiler_driver.py
index 1d8756b..cace0f0 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 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 @@
         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
+    )