diff options
author | Johan Alfven <johan.alfven@arm.com> | 2023-03-09 08:36:10 +0100 |
---|---|---|
committer | Fredrik Svedberg <fredrik.svedberg@arm.com> | 2023-03-16 16:12:36 +0000 |
commit | 126558e26df26830c2d331ec0041dc9a4f1a0d38 (patch) | |
tree | 597de0b202cfcd3950faf9c7f4e71e56ed0d867d /ethosu/vela/tensor.py | |
parent | a5e1b6224d8436365e7f0bdb0afef060423fba57 (diff) | |
download | ethos-u-vela-126558e26df26830c2d331ec0041dc9a4f1a0d38.tar.gz |
MLBEDSW-7352: Refactoring move_constant_data
Refactoring move_constant_data in the scheduler. The use case currently
only work for LUT tensor, so simplifying the logic. In order to make it
work for other tensors one would also have to take into consideration
memory usage when building cascades and also the
use_fast_storage_for_feature_maps would be effected.
Change-Id: Ic8de53b65a2c17d34515002d7f184d0ab1830222
Signed-off-by: Johan Alfven <johan.alfven@arm.com>
Diffstat (limited to 'ethosu/vela/tensor.py')
-rw-r--r-- | ethosu/vela/tensor.py | 7 |
1 files changed, 3 insertions, 4 deletions
diff --git a/ethosu/vela/tensor.py b/ethosu/vela/tensor.py index 008cd05e..86306cad 100644 --- a/ethosu/vela/tensor.py +++ b/ethosu/vela/tensor.py @@ -501,10 +501,9 @@ class Tensor: return res - def clone_into_fast_storage(self, arch) -> "Tensor": - res = self.clone(suffix="_fast_storage") - res.mem_area = arch.fast_storage_mem_area - res.mem_type = MemType.Scratch_fast + def clone_into_shram(self, arch) -> "Tensor": + res = self.clone(suffix="_shram") + res.mem_area = MemArea.Shram res.src_tensor = self return res |