diff options
author | Tim Hall <tim.hall@arm.com> | 2022-03-03 17:50:52 +0000 |
---|---|---|
committer | Tim Hall <tim.hall@arm.com> | 2022-05-17 20:06:24 +0100 |
commit | c1be0873d9e28a21c7873793da896e6dd576292f (patch) | |
tree | 8d8392a7ae2a821e6dbc9f048343f9ccbe9e5828 /ethosu/vela/scheduler.py | |
parent | 3dae1b6088a469f1073222bf249d17c8cdf39dbf (diff) | |
download | ethos-u-vela-c1be0873d9e28a21c7873793da896e6dd576292f.tar.gz |
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
Diffstat (limited to 'ethosu/vela/scheduler.py')
-rw-r--r-- | ethosu/vela/scheduler.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/ethosu/vela/scheduler.py b/ethosu/vela/scheduler.py index bc058762..d65f1dc5 100644 --- a/ethosu/vela/scheduler.py +++ b/ethosu/vela/scheduler.py @@ -1006,7 +1006,7 @@ class Scheduler: 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 @@ class Scheduler: 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] |