commit 7b8823f0e27ec115a2e151ebc0382a49fce36894
author Charles Xu <charles.xu@arm.com> Fri May 29 13:53:10 2020 +0200
committer Tim Hall <tim.hall@arm.com> Thu Jun 18 17:53:52 2020 +0100
tree 551741b51aa1f184a44a366594b450282b5471c5
parent 332a704c0dddfc31eef3c3a6119b1800da2e49fe

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

ethosu/vela/vela.py

diff --git a/ethosu/vela/vela.py b/ethosu/vela/vela.py
index 8dda860..20bc525 100644
--- a/ethosu/vela/vela.py
+++ b/ethosu/vela/vela.py
@@ -246,6 +246,13 @@
             "(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 @@
         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()