MLBEDSW-2306 Added more supported mem-cfgs
Additional supported memory configurations:
-Permanent_storage = DRAM
-Tensor arena either in DRAM or SRAM
Signed-off-by: Patrik Gustavsson <patrik.gustavsson@arm.com>
Change-Id: I20beb7151e306bfdba540e7c0b2a7b478b4d94e1
diff --git a/ethosu/vela/scheduler.py b/ethosu/vela/scheduler.py
index 0b59431..be104b8 100644
--- a/ethosu/vela/scheduler.py
+++ b/ethosu/vela/scheduler.py
@@ -38,6 +38,7 @@
from .shared_buffer_allocation import find_block_configs_suitable_for_pass_and_shared_buffer
from .shared_buffer_allocation import shared_buffer_allocation_for_pass_and_block_config
from .tensor import MemArea
+from .tensor import MemType
from .tensor import TensorFormat
from .tensor import TensorPurpose
from .tensor import TensorSubPurpose
@@ -833,6 +834,7 @@
for rewrite_op, tens, sub_purpose, param_a, param_b, ps in strat.rewrite_list:
if rewrite_op == SchedulerRewrite.ChangeTensorSubPurpose:
tens.mem_area = self.arch.fast_storage_mem_area
+ tens.mem_type = MemType.Scratch_fast
tens.set_new_sub_purpose(sub_purpose, param_a, param_b)
else:
assert 0, "unknown rewrite_op " + str(rewrite_op)