diff options
author | Andreas Nevalainen <andreas.nevalainen@arm.com> | 2020-10-28 15:42:08 +0100 |
---|---|---|
committer | patrik.gustavsson <patrik.gustavsson@arm.com> | 2020-11-11 08:34:16 +0000 |
commit | 897cc14968e017b1f48f376f7f7cefc515c5fe88 (patch) | |
tree | 27d17a59f1529c1ed0b1cc59e60438ee7f5d29d2 /ethosu/vela/vela.py | |
parent | 73320a48dfa711f5938b0e3d8e03b9858558b899 (diff) | |
download | ethos-u-vela-897cc14968e017b1f48f376f7f7cefc515c5fe88.tar.gz |
MLBEDSW-3222: Bias tensors in fast storage
For IFM streamed cascades bias tensors are read several times.
Moves these tensors to fast storage and add DMA commands.
Change-Id: I630f6275986c1b5e3f126c925b11e22500fb1128
Signed-off-by: Andreas Nevalainen <andreas.nevalainen@arm.com>
Diffstat (limited to 'ethosu/vela/vela.py')
-rw-r--r-- | ethosu/vela/vela.py | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/ethosu/vela/vela.py b/ethosu/vela/vela.py index 6d54187c..4b43751a 100644 --- a/ethosu/vela/vela.py +++ b/ethosu/vela/vela.py @@ -152,6 +152,9 @@ def main(args=None): "--show-cpu-operations", action="store_true", help="Show the operations that fall back to the CPU" ) parser.add_argument( + "--keep-scale-placement", action="store_true", help="Keep scale tensors memory placement during scheduling" + ) + parser.add_argument( "--cascading", type=ast.literal_eval, default=True, @@ -311,6 +314,7 @@ def main(args=None): use_ifm_streaming=args.ifm_streaming, pareto_metric=args.pareto_metric, use_nhcwb16_between_cascaded_passes=args.nhcwb16_between_cascaded_passes, + keep_scale_placement=args.keep_scale_placement, ) model_reader_options = model_reader.ModelReaderOptions() |