diff options
author | Charles Xu <charles.xu@arm.com> | 2020-05-29 13:53:10 +0200 |
---|---|---|
committer | Tim Hall <tim.hall@arm.com> | 2020-06-18 17:53:52 +0100 |
commit | 7b8823f0e27ec115a2e151ebc0382a49fce36894 (patch) | |
tree | 551741b51aa1f184a44a366594b450282b5471c5 /ethosu/vela/vela.py | |
parent | 332a704c0dddfc31eef3c3a6119b1800da2e49fe (diff) | |
download | ethos-u-vela-7b8823f0e27ec115a2e151ebc0382a49fce36894.tar.gz |
MLBEDSW-2370: Add CLI option for NHCWB16
Make it configurable for using NHCWB16 between
cascaded passes.
Signed-off-by: Charles Xu <charles.xu@arm.com>
Change-Id: I259cdaa424d11ea38f17e671490ad1e630bbae44
Diffstat (limited to 'ethosu/vela/vela.py')
-rw-r--r-- | ethosu/vela/vela.py | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/ethosu/vela/vela.py b/ethosu/vela/vela.py index 8dda8606..20bc525b 100644 --- a/ethosu/vela/vela.py +++ b/ethosu/vela/vela.py @@ -246,6 +246,13 @@ def main(args=None): "(default: %(default)s)" ), ) + parser.add_argument( + "--nhcwb16-between-cascaded-passes", + type=ast.literal_eval, + default=True, + choices=[True, False], + help="Control if NHCWB16 or NHWC should be used in between cascaded passes (default: %(default)s)", + ) args = parser.parse_args(args=args) @@ -302,6 +309,7 @@ def main(args=None): verbose_pareto_frontier_schedules=args.verbose_pareto_frontier_schedules, use_ifm_streaming=args.ifm_streaming, pareto_metric=args.pareto_metric, + use_nhcwb16_between_cascaded_passes=args.nhcwb16_between_cascaded_passes, ) model_reader_options = model_reader.ModelReaderOptions() |