diff options
author | Patrik Gustavsson <patrik.gustavsson@arm.com> | 2021-02-11 08:36:12 +0100 |
---|---|---|
committer | Patrik Gustavsson <patrik.gustavsson@arm.com> | 2021-02-11 15:02:22 +0100 |
commit | 2446e59a9083f36f85beb88fdec6379d90b85cad (patch) | |
tree | 11ec29acb69d8f9c49d4a339d0fb7686f71d9b2a /ethosu/vela/graph_optimiser.py | |
parent | e3b1b91c450129308a3a1d466a2f2876a5a872b1 (diff) | |
download | ethos-u-vela-2446e59a9083f36f85beb88fdec6379d90b85cad.tar.gz |
MLBEDSW-3774 Fix avoid cascading for spilling
Fix avoid cascading for spilling.
Signed-off-by: Patrik Gustavsson <patrik.gustavsson@arm.com>
Change-Id: If86189bd1566eaa14387dfc2c02e3324ea6c184e
Diffstat (limited to 'ethosu/vela/graph_optimiser.py')
-rw-r--r-- | ethosu/vela/graph_optimiser.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/ethosu/vela/graph_optimiser.py b/ethosu/vela/graph_optimiser.py index 50368b86..f5006c6f 100644 --- a/ethosu/vela/graph_optimiser.py +++ b/ethosu/vela/graph_optimiser.py @@ -109,6 +109,7 @@ def rewrite_concat_ops(op, arch): DebugDatabase.add_optimised(op, avgpool_op) avgpool_op.ifm_shapes.append(op.ifm_shapes[idx]) avgpool_op.ofm_shapes.append(op.ofm_shapes[0]) + avgpool_op.memory_function = Op.ConcatSliceWrite assert ofm.shape[axis] == offset # If axis corresponds to C-dimension, NHCWB16 can only be used in the output if all the concat_start's are a |