MLBEDSW-2549 Crash with incorrect strided slice op
A valid strided slice should have (positive) non-zero elements
when you do "end - begin"
When encountering an invalid strided slice, vela asserted.
This now checks that it is valid and wont claim support if it isnt.
Signed-off-by: Michael McGeagh <michael.mcgeagh@arm.com>
Change-Id: I33ef118bd6a31ac78c680acb5229ff31b0809d6a
diff --git a/ethosu/vela/supported_operators.py b/ethosu/vela/supported_operators.py
index b585323..43ba36f 100644
--- a/ethosu/vela/supported_operators.py
+++ b/ethosu/vela/supported_operators.py
@@ -330,6 +330,9 @@
# check stride size
if len(op.inputs) > 3 and any(stride != 1 for stride in op.inputs[3].values):
return False
+ # check "end - begin" doesnt result in any zero or negative elements
+ if any((end - begin) <= 0 for begin, end in zip(op.inputs[1].values, op.inputs[2].values)):
+ return False
# check ellipsis_mask
if op.attrs["ellipsis_mask"] != 0:
return False