commit 27d36f003d35413beb51c1de8f33259ddeca7543
author Andreas Nevalainen <andreas.nevalainen@arm.com> Thu Nov 19 11:27:50 2020 +0100
committer patrik.gustavsson <patrik.gustavsson@arm.com> Mon Nov 23 11:51:08 2020 +0000
tree a386f0a183ccd2896d6630146bcac554a1828feb
parent 43f8f6424cb942f27599258607ea36c9a852f5ef

MLBEDSW-3468: Move of scale tensors to SRAM after weight compressor

After weight compressor weights has correct sizes. Placing move of scale
tensors after weight compressor gives more accurate estimate of available
SRAM for scale tensors.

Change-Id: I4571780180778ef43e943c4e98048e17d6f33580
Signed-off-by: Andreas Nevalainen <andreas.nevalainen@arm.com>

ethosu/vela/compiler_driver.py

diff --git a/ethosu/vela/compiler_driver.py b/ethosu/vela/compiler_driver.py
index a2b20e4..7b1ea21 100644
--- a/ethosu/vela/compiler_driver.py
+++ b/ethosu/vela/compiler_driver.py
@@ -174,6 +174,9 @@
     # block config, and calc and pack the scales and biases
     weight_compressor.update_pass_weight_and_scale_tensors(nng, arch)
 
+    if not scheduler_options.keep_scale_placement:
+        scheduler.move_scales_to_fast_storage(nng, arch)
+
     # LiveRanges for constant tensors for all Npu subgraphs
     permanent_storage = arch.permanent_storage_mem_area
     lr_graph_flash = live_range.LiveRangeGraph()