diff options
author | Patrik Gustavsson <patrik.gustavsson@arm.com> | 2021-02-09 15:38:46 +0100 |
---|---|---|
committer | Patrik Gustavsson <patrik.gustavsson@arm.com> | 2021-02-11 14:38:56 +0100 |
commit | e3b1b91c450129308a3a1d466a2f2876a5a872b1 (patch) | |
tree | 77c6fe5cc8836dc3aac505efdbf78cee433bd398 /ethosu/vela/pass_packing.py | |
parent | 138d47f5a3e87d294b3714ae799ccad8ac9111bd (diff) | |
download | ethos-u-vela-e3b1b91c450129308a3a1d466a2f2876a5a872b1.tar.gz |
MLBEDSW-3774 Remove SplitSliceRead
Removed SplitSliceRead from subgraph during
graph optimisation.
Signed-off-by: Patrik Gustavsson <patrik.gustavsson@arm.com>
Change-Id: I9315d4c2a6767828dd2b4e66823d73b10ebee99c
Diffstat (limited to 'ethosu/vela/pass_packing.py')
-rw-r--r-- | ethosu/vela/pass_packing.py | 16 |
1 files changed, 2 insertions, 14 deletions
diff --git a/ethosu/vela/pass_packing.py b/ethosu/vela/pass_packing.py index b52b159e..281c0932 100644 --- a/ethosu/vela/pass_packing.py +++ b/ethosu/vela/pass_packing.py @@ -1,4 +1,4 @@ -# Copyright (C) 2020 Arm Limited or its affiliates. All rights reserved. +# Copyright (C) 2020-2021 Arm Limited or its affiliates. All rights reserved. # # SPDX-License-Identifier: Apache-2.0 # @@ -42,8 +42,6 @@ class PassFlags(enum.Flag): PostFusingLimited = 8192 -npu_pre_ops = set((Op.SplitSliceRead,)) - mac_main_ops = set( ( # convolutions @@ -148,16 +146,6 @@ test_sequence = [ ), ( # ops_set - npu_pre_ops, - # incompatible_pack_flags - PassFlags.Cpu | PassFlags.MemoryOnly, - # flags_to_set - PassFlags.Npu | PassFlags.Mac | PassFlags.Pre | PassFlags.ElementWise, - # flags_to_clear - PassFlags.Empty, - ), - ( - # ops_set npu_dma_ops, # incompatible_pack_flags PassFlags.Cpu | PassFlags.MemoryOnly, @@ -437,7 +425,7 @@ def pack_into_passes(nng, arch, verbose_packing=False): visit_op(op, tens) 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): + if any(op.type in (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 = op_list[0] inp = op.inputs[0] |