commit c1be0873d9e28a21c7873793da896e6dd576292f
author Tim Hall <tim.hall@arm.com> Thu Mar 03 17:50:52 2022 +0000
committer Tim Hall <tim.hall@arm.com> Tue May 17 20:06:24 2022 +0100
tree 8d8392a7ae2a821e6dbc9f048343f9ccbe9e5828
parent 3dae1b6088a469f1073222bf249d17c8cdf39dbf

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

ethosu/vela/scheduler.py

diff --git a/ethosu/vela/scheduler.py b/ethosu/vela/scheduler.py
index bc05876..d65f1dc 100644
--- a/ethosu/vela/scheduler.py
+++ b/ethosu/vela/scheduler.py
@@ -1006,7 +1006,7 @@
         options: SchedulerOptions,
     ):
         default_schedule = self.sg.schedule
-        npu_performance.calc_new_performance_for_network(self.nng, self.arch)
+        npu_performance.calc_new_performance_for_network(self.nng, self.arch, None, False)
         default_tot_cycles = self.nng.cycles[npu_performance.PassCycles.Total]
         default_dram_cycles = self.nng.cycles[npu_performance.PassCycles.DramAccess]
 
@@ -1069,7 +1069,7 @@
         self.apply_schedule(self.sg.schedule)
         self.use_fast_storage_for_feature_maps(self.sg.schedule, options.optimization_sram_limit)
 
-        npu_performance.calc_new_performance_for_network(self.nng, self.arch)
+        npu_performance.calc_new_performance_for_network(self.nng, self.arch, None, False)
         new_tot_cycles = self.nng.cycles[npu_performance.PassCycles.Total]
         new_dram_cycles = self.nng.cycles[npu_performance.PassCycles.DramAccess]