diff options
author | Patrik Gustavsson <patrik.gustavsson@arm.com> | 2020-05-27 09:15:11 +0200 |
---|---|---|
committer | Patrik Gustavsson <patrik.gustavsson@arm.com> | 2020-06-25 11:42:56 +0200 |
commit | eca2e95e1fea150d8a942f8b5f0a4d9d7aefebc1 (patch) | |
tree | 438b385f1ded3c18c3b84d2204a57c39be6be34a /ethosu/vela/scheduler.py | |
parent | eec4e50e19cb5522640eae5fd4566917dc2a7b9d (diff) | |
download | ethos-u-vela-eca2e95e1fea150d8a942f8b5f0a4d9d7aefebc1.tar.gz |
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
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 0b594310..be104b88 100644 --- a/ethosu/vela/scheduler.py +++ b/ethosu/vela/scheduler.py @@ -38,6 +38,7 @@ from .operation import NpuBlockType 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 @@ class DynamicProgrammingScheduler: 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) |