diff options
| author | Jacob Bohlin <jacob.bohlin@arm.com> | 2020-08-17 09:42:35 +0200 |
|---|---|---|
| committer | tim.hall <tim.hall@arm.com> | 2020-08-19 17:43:54 +0000 |
| commit | fb858733d9b5b937fea9222e407b0a780e0e48c9 (patch) | |
| tree | 84a7903ab73733fe2cd27e6a7263b5bca82e355c | |
| parent | d2a972d8b400a84062793eafe5dfc65e939804b7 (diff) | |
| download | ethos-u-vela-fb858733d9b5b937fea9222e407b0a780e0e48c9.tar.gz | |
MLBEDSW-2728: Only insert primary op for NPU ops
Signed-off-by: Jacob Bohlin <jacob.bohlin@arm.com>
Change-Id: I04f299e2d3319113fedf2fa401b88bae64fea66d
| -rw-r--r-- | ethosu/vela/pass_packing.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/ethosu/vela/pass_packing.py b/ethosu/vela/pass_packing.py index 7b69e35d..a4caf0c0 100644 --- a/ethosu/vela/pass_packing.py +++ b/ethosu/vela/pass_packing.py @@ -442,10 +442,10 @@ def pack_into_passes(nng, arch, verbose_packing=False): for op in reversed(tens.ops): visit_op(op, tens) - def create_primary_op(ops_list): - if any(op.type in (npu_pre_ops | npu_post_ops | npu_post_fuse_limited_ops) for op in ops_list): + def create_primary_op(op_list): + if any(op.type in (npu_pre_ops | npu_post_ops | npu_post_fuse_limited_ops) and op.run_on_npu for op in op_list): # Configure a 1x1 AvgPool and attach the op onto it - op = ops_list[0] + op = op_list[0] inp = op.inputs[0] avgpool_name = op.name + "_avgpool" avgpool_op = Operation("AvgPool", avgpool_name) @@ -466,7 +466,7 @@ def pack_into_passes(nng, arch, verbose_packing=False): avgpool_op.set_output_tensor(avgpool_out) op.inputs[0] = avgpool_out - ops_list.insert(0, avgpool_op) + op_list.insert(0, avgpool_op) return avgpool_op |