From 69aadd052588eb53a257e8f7431ed858161b3286 Mon Sep 17 00:00:00 2001 From: Diqing Zhong Date: Tue, 8 Dec 2020 13:08:48 +0100 Subject: Vela: bandwidth calculation improvements - Combine conv and vector_product calculation - Remove internal bandwidth - Remove blocks and hw_macs from report - Use scaled_bws for cycle estimation Related to: MLBEDSW-3598 Change-Id: I1927a8311ec563f68115e0f2ed077806b86fd717 Signed-off-by: Diqing Zhong --- ethosu/vela/scheduler.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) (limited to 'ethosu/vela/scheduler.py') diff --git a/ethosu/vela/scheduler.py b/ethosu/vela/scheduler.py index 977eb58e..2c10640b 100644 --- a/ethosu/vela/scheduler.py +++ b/ethosu/vela/scheduler.py @@ -32,7 +32,6 @@ from .nn_graph import SchedulerRewrite from .nn_graph import SchedulingStrategy from .npu_performance import make_bandwidth_array from .npu_performance import make_cycles_array -from .npu_performance import make_macs_array from .npu_performance import make_metrics_arrays from .npu_performance import PassCycles from .numeric_util import full_shape @@ -108,7 +107,7 @@ class Strategy: return False if (self.bws != other.bws).any(): return False - if (self.macs != other.macs).any(): + if self.macs != other.macs: return False if (self.cycles != other.cycles).any(): return False @@ -211,7 +210,7 @@ class StrategySet: empty_strategy = Strategy( - SchedulingStrategy.Unknown, None, [], [], [], make_bandwidth_array(), make_macs_array(), make_cycles_array(), 0 + SchedulingStrategy.Unknown, None, [], [], [], make_bandwidth_array(), 0, make_cycles_array(), 0 ) INFINITY = 1e30 -- cgit v1.2.1