commit 94292fe32c34357a8935a42c77b759a499eb0db9
author Patrik Gustavsson <patrik.gustavsson@arm.com> Thu Sep 02 08:22:58 2021 +0200
committer patrik.gustavsson <patrik.gustavsson@arm.com> Tue Sep 07 11:44:49 2021 +0000
tree f9169dff2dd2b8afa6da1d5532603fe440cc6b39
parent fad720480c3b115664ed5d0dc6028c5020005cf7

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

ethosu/vela/scheduler.py

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(