MLBEDSW-5160 Fix constant data move to fast storage
Additional check added for when constant data can be moved
to fast storage.
Do not move constant data for concat.
Signed-off-by: Patrik Gustavsson <patrik.gustavsson@arm.com>
Change-Id: Ib8b5fd1483ee9fabe48e9874a5723af9b7c5231a
diff --git a/ethosu/vela/scheduler.py b/ethosu/vela/scheduler.py
index f96b773..e4543e3 100644
--- a/ethosu/vela/scheduler.py
+++ b/ethosu/vela/scheduler.py
@@ -959,10 +959,12 @@
and self.arch.permanent_storage_mem_area != self.arch.feature_map_storage_mem_area
) or tens.purpose == TensorPurpose.LUT:
if tens.purpose == TensorPurpose.LUT or (
+ # For elementwise broadcast
tens.purpose == TensorPurpose.FeatureMap
and sched_op.op_type.is_binary_elementwise_op()
and tens.shape != []
and sched_op.ifm.shape != sched_op.ofm.shape
+ and parent_op.write_shape is None
and tens.storage_size() > max_ifm_shram_avail
):
only_vector_product_consumers = all(