diff options
author | Patrik Gustavsson <patrik.gustavsson@arm.com> | 2020-08-24 16:26:11 +0200 |
---|---|---|
committer | tim.hall <tim.hall@arm.com> | 2020-08-26 09:37:01 +0000 |
commit | 90831bc18d45008b703e59aad0594026beb7da82 (patch) | |
tree | 6b647996160ba8ecd037da1e15c91f55d4b0f665 /ethosu/vela/compiler_driver.py | |
parent | d47cc2777f3a3f371958c30a3c1880f692b0b0a2 (diff) | |
download | ethos-u-vela-90831bc18d45008b703e59aad0594026beb7da82.tar.gz |
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
Diffstat (limited to 'ethosu/vela/compiler_driver.py')
-rw-r--r-- | ethosu/vela/compiler_driver.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/ethosu/vela/compiler_driver.py b/ethosu/vela/compiler_driver.py index 5e9e38fb..94900ad5 100644 --- a/ethosu/vela/compiler_driver.py +++ b/ethosu/vela/compiler_driver.py @@ -212,7 +212,9 @@ def compiler_driver(nng, arch, options, scheduler_options): if root_sg is not None and (arch.feature_map_storage_mem_area != arch.fast_storage_mem_area): if root_sg.memory_used_per_type.get(MemType.Scratch_fast, 0) > arch.sram_size: raise VelaError( - "Sram limit {} bytes, has been exceeded by the scratch fast tensor {} bytes".format( + "Sram limit {} bytes, has been exceeded by the scratch fast tensor {} bytes. " + "Increasing the value of --weight-estimation-scaling may help to resolve the issue. " + "See OPTIONS.md for more information.".format( arch.sram_size, root_sg.memory_used_per_type.get(MemType.Scratch_fast, 0) ) ) |