Skip to content

Commit ac3d172

Browse files
committed
[NE16] pass --enable-3x3 from deeployRunner to generateNetwork
The deeployRunner parsed --enable-3x3 but never forwarded it to generateNetwork.py's gen_args, so NE16Engine.enable3x3 stayed False and DW/Dense 3x3 convs silently fell back to the cluster.
1 parent ee2f1b9 commit ac3d172

1 file changed

Lines changed: 3 additions & 0 deletions

File tree

DeeployTest/testUtils/deeployRunner.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -239,6 +239,9 @@ def create_config_from_args(args: argparse.Namespace,
239239
if hasattr(args, 'plotMemAlloc') and args.plotMemAlloc:
240240
gen_args_list.append("--plotMemAlloc")
241241

242+
if getattr(args, 'enable_3x3', False):
243+
gen_args_list.append("--enable-3x3")
244+
242245
if not tiling and getattr(args, 'profileUntiled', False):
243246
gen_args_list.append("--profileUntiled")
244247

0 commit comments

Comments
 (0)