diff options
author | Andreas Nevalainen <andreas.nevalainen@arm.com> | 2020-11-19 11:27:50 +0100 |
---|---|---|
committer | patrik.gustavsson <patrik.gustavsson@arm.com> | 2020-11-23 11:51:08 +0000 |
commit | 27d36f003d35413beb51c1de8f33259ddeca7543 (patch) | |
tree | a386f0a183ccd2896d6630146bcac554a1828feb /ethosu/vela/compiler_driver.py | |
parent | 43f8f6424cb942f27599258607ea36c9a852f5ef (diff) | |
download | ethos-u-vela-27d36f003d35413beb51c1de8f33259ddeca7543.tar.gz |
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>
Diffstat (limited to 'ethosu/vela/compiler_driver.py')
-rw-r--r-- | ethosu/vela/compiler_driver.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/ethosu/vela/compiler_driver.py b/ethosu/vela/compiler_driver.py index a2b20e47..7b1ea213 100644 --- a/ethosu/vela/compiler_driver.py +++ b/ethosu/vela/compiler_driver.py @@ -174,6 +174,9 @@ def compiler_driver(nng, arch, options, scheduler_options): # 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() |