commit 2446e59a9083f36f85beb88fdec6379d90b85cad
author Patrik Gustavsson <patrik.gustavsson@arm.com> Thu Feb 11 08:36:12 2021 +0100
committer Patrik Gustavsson <patrik.gustavsson@arm.com> Thu Feb 11 15:02:22 2021 +0100
tree 11ec29acb69d8f9c49d4a339d0fb7686f71d9b2a
parent e3b1b91c450129308a3a1d466a2f2876a5a872b1

MLBEDSW-3774 Fix avoid cascading for spilling

Fix avoid cascading for spilling.

Signed-off-by: Patrik Gustavsson <patrik.gustavsson@arm.com>
Change-Id: If86189bd1566eaa14387dfc2c02e3324ea6c184e

ethosu/vela/graph_optimiser.py

diff --git a/ethosu/vela/graph_optimiser.py b/ethosu/vela/graph_optimiser.py
index 50368b8..f5006c6 100644
--- a/ethosu/vela/graph_optimiser.py
+++ b/ethosu/vela/graph_optimiser.py
@@ -109,6 +109,7 @@
         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