diff options
| author | Diqing Zhong <diqing.zhong@arm.com> | 2020-12-08 13:08:48 +0100 |
|---|---|---|
| committer | Diqing Zhong <diqing.zhong@arm.com> | 2020-12-09 15:20:38 +0100 |
| commit | 69aadd052588eb53a257e8f7431ed858161b3286 (patch) | |
| tree | 4489ef3ad631a71a057b36fefa83afd83a93a8e7 /ethosu/vela/scheduler.py | |
| parent | 2fa40ae3e0e17c899e847c8ad8decd1ec0d9bfcd (diff) | |
| download | ethos-u-vela-69aadd052588eb53a257e8f7431ed858161b3286.tar.gz | |
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 <diqing.zhong@arm.com>
Diffstat (limited to 'ethosu/vela/scheduler.py')
| -rw-r--r-- | ethosu/vela/scheduler.py | 5 |
1 files changed, 2 insertions, 3 deletions
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 |