commit 90831bc18d45008b703e59aad0594026beb7da82
author Patrik Gustavsson <patrik.gustavsson@arm.com> Mon Aug 24 16:26:11 2020 +0200
committer tim.hall <tim.hall@arm.com> Wed Aug 26 09:37:01 2020 +0000
tree 6b647996160ba8ecd037da1e15c91f55d4b0f665
parent d47cc2777f3a3f371958c30a3c1880f692b0b0a2

MLBED-2822 Added CLI-opt for weight size est.

Added --weight-estimation-scaling, which enables
additional scaling of weight compression scale estimate.

Signed-off-by: Patrik Gustavsson <patrik.gustavsson@arm.com>
Change-Id: Idcda41257f44901d3a3f345341e07fb1ae8585a9

ethosu/vela/architecture_features.py

diff --git a/ethosu/vela/architecture_features.py b/ethosu/vela/architecture_features.py
index 5453f2c..8b968a3 100644
--- a/ethosu/vela/architecture_features.py
+++ b/ethosu/vela/architecture_features.py
@@ -184,6 +184,7 @@
         global_memory_clock_scale,
         max_blockdep,
         softmax_support,
+        weight_estimation_scaling,
     ):
         accelerator_config = accelerator_config.lower()
         self.vela_config = vela_config
@@ -215,6 +216,7 @@
             )
 
         self.max_blockdep = max_blockdep
+        self.weight_estimation_scaling = weight_estimation_scaling
 
         dpu_min_height = accel_config.ofm_ublock.height
         dpu_min_width = accel_config.ofm_ublock.width