diff options
author | Patrik Gustavsson <patrik.gustavsson@arm.com> | 2021-09-02 08:22:58 +0200 |
---|---|---|
committer | patrik.gustavsson <patrik.gustavsson@arm.com> | 2021-09-07 11:44:49 +0000 |
commit | 94292fe32c34357a8935a42c77b759a499eb0db9 (patch) | |
tree | f9169dff2dd2b8afa6da1d5532603fe440cc6b39 /ethosu/vela/scheduler.py | |
parent | fad720480c3b115664ed5d0dc6028c5020005cf7 (diff) | |
download | ethos-u-vela-94292fe32c34357a8935a42c77b759a499eb0db9.tar.gz |
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
Diffstat (limited to 'ethosu/vela/scheduler.py')
-rw-r--r-- | ethosu/vela/scheduler.py | 2 |
1 files changed, 2 insertions, 0 deletions
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( |