commit dda21afda93f3732491efdcf89af2b14396c683f
author Dwight Lidman <dwight.lidman@arm.com> Wed Nov 11 15:44:57 2020 +0100
committer patrik.gustavsson <patrik.gustavsson@arm.com> Wed Nov 11 15:58:08 2020 +0000
tree 872aaf75b7eb005b1affb01f56b9a48de093dc48
parent 7aa5aa4b520789331905ce2d35fe3d82dd5ac760

MLBEDSW-3463: StridedSlice fixup function causes infinite recursion

This commit reverts a control flow path where
already modified StridedSlice operators are
left untouched.
If not, Vela would recurse infinitely and crash.

Signed-off-by: Dwight Lidman <dwight.lidman@arm.com>
Change-Id: Iaf3ae916325bedd3dd1edd3395fb4a9ecf832590

ethosu/vela/graph_optimiser.py

diff --git a/ethosu/vela/graph_optimiser.py b/ethosu/vela/graph_optimiser.py
index 7304630..899da07 100644
--- a/ethosu/vela/graph_optimiser.py
+++ b/ethosu/vela/graph_optimiser.py
@@ -456,6 +456,9 @@
 
             assert len(tens.shape) == (len(op.inputs[0].shape) + n)
             op.attrs["new_axis_mask"] = 0
+        else:
+            # Equal Rank StridedSlice, no need to insert reshape
+            return tens
 
         # Construct 1 shape tensor to be used by all inserted reshape ops
         new_shape_tens = create_const_tensor(op.name + "_reshape_shape", [1], DataType.int32, tens.shape)