diff options
author | Tim Hall <tim.hall@arm.com> | 2022-05-19 15:29:23 +0100 |
---|---|---|
committer | tim.hall <tim.hall@arm.com> | 2022-05-19 15:56:19 +0000 |
commit | 8bc7a652607a771e234fda6b05275542ff0fc072 (patch) | |
tree | 5ac2839b0bbe9c964a803f4b047e1d2795f1326e | |
parent | eb5a4a857cca3ae18928268adfcffe91efd48fd9 (diff) | |
download | ethos-u-vela-8bc7a652607a771e234fda6b05275542ff0fc072.tar.gz |
MLBEDSW-6296: improvement_dram can become NaN
- Problem is due to a divide by zero
- Fix is simply to detect and assign zero. This could also affect
improvement_sram
Signed-off-by: Tim Hall <tim.hall@arm.com>
Change-Id: I29a67710a17ef22656fb5ecfe9476953ffa5533d
-rw-r--r-- | ethosu/vela/scheduler.py | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/ethosu/vela/scheduler.py b/ethosu/vela/scheduler.py index d65f1dc5..0b79402a 100644 --- a/ethosu/vela/scheduler.py +++ b/ethosu/vela/scheduler.py @@ -1073,8 +1073,12 @@ class Scheduler: new_tot_cycles = self.nng.cycles[npu_performance.PassCycles.Total] new_dram_cycles = self.nng.cycles[npu_performance.PassCycles.DramAccess] - improvement_tot = round((default_tot_cycles - new_tot_cycles) / default_tot_cycles, 2) - improvement_dram = round((default_dram_cycles - new_dram_cycles) / default_dram_cycles, 2) + improvement_tot = ( + round((default_tot_cycles - new_tot_cycles) / default_tot_cycles, 2) if default_tot_cycles != 0 else 0 + ) + improvement_dram = ( + round((default_dram_cycles - new_dram_cycles) / default_dram_cycles, 2) if default_dram_cycles != 0 else 0 + ) # Compare both total and dram improvement if not (improvement_tot >= 0 and improvement_dram > 0): |