From 94292fe32c34357a8935a42c77b759a499eb0db9 Mon Sep 17 00:00:00 2001 From: Patrik Gustavsson Date: Thu, 2 Sep 2021 08:22:58 +0200 Subject: 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 Change-Id: Ib8b5fd1483ee9fabe48e9874a5723af9b7c5231a --- ethosu/vela/scheduler.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/ethosu/vela/scheduler.py b/ethosu/vela/scheduler.py index f96b7732..e4543e3a 100644 --- a/ethosu/vela/scheduler.py +++ b/ethosu/vela/scheduler.py @@ -959,10 +959,12 @@ class Scheduler: 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( -- cgit v1.2.1