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>
diff --git a/ethosu/vela/test/test_lut.py b/ethosu/vela/test/test_lut.py
index 58e72bb..e52b489 100644
--- a/ethosu/vela/test/test_lut.py
+++ b/ethosu/vela/test/test_lut.py
@@ -36,7 +36,7 @@
random.seed(key)
values = random.choices(range(256), k=256)
lut_tensor = create_const_tensor(op.name + "_lut", [1, 1, 1, 256], DataType.uint8, values, TensorPurpose.LUT)
- scratch_lut_tensor = lut_tensor.clone_into_fast_storage(arch)
+ scratch_lut_tensor = lut_tensor.clone_into_shram(arch)
op.set_activation_lut(scratch_lut_tensor)
@@ -44,7 +44,7 @@
random.seed(key)
values = random.choices(range(256), k=256)
lut_tensor = create_const_tensor(op.name + "_lut", [1, 1, 1, 256], DataType.int32, values, TensorPurpose.LUT)
- scratch_lut_tensor = lut_tensor.clone_into_fast_storage(arch)
+ scratch_lut_tensor = lut_tensor.clone_into_shram(arch)
op.set_activation_lut(scratch_lut_tensor)
@@ -52,7 +52,7 @@
random.seed(key)
values = random.choices(range(512), k=512)
lut_tensor = create_const_tensor(op.name + "_lut", [1, 1, 1, 512], DataType.int32, values, TensorPurpose.LUT)
- scratch_lut_tensor = lut_tensor.clone_into_fast_storage(arch)
+ scratch_lut_tensor = lut_tensor.clone_into_shram(arch)
op.set_activation_lut(scratch_lut_tensor)