diff options
| author | Tim Hall <tim.hall@arm.com> | 2020-11-01 20:59:36 +0000 |
|---|---|---|
| committer | Tim Hall <tim.hall@arm.com> | 2020-11-20 12:55:47 +0000 |
| commit | 1bd531dec0b4eb745fb8856d14c1aba2b8a73026 (patch) | |
| tree | a0265a0accd2395277fe88be27164d09541abc7f /ethosu/vela/high_level_command_to_npu_op.py | |
| parent | c8a73868d40cf63380f634baeb51aa7aa993fc0c (diff) | |
| download | ethos-u-vela-1bd531dec0b4eb745fb8856d14c1aba2b8a73026.tar.gz | |
MLBEDSW-3249: Vela config file examples
- Added sample vela.ini config file
- Changed vela config format, split into system config and memory mode
- Removed unused CPU cycle performance estimation
- Added new CLI options for --memory-mode and --verbose-config
- Changed CLI option --config to take multiple files
- Removed CLI option --global-memory-clock-scales
- Changed error helper functions to raise a VelaError exception
- Refactored to create a new is_spilling_enabled function
Signed-off-by: Tim Hall <tim.hall@arm.com>
Change-Id: I27c41577e37a3859edb9524cd99784be10ef0a0d
Diffstat (limited to 'ethosu/vela/high_level_command_to_npu_op.py')
| -rw-r--r-- | ethosu/vela/high_level_command_to_npu_op.py | 23 |
1 files changed, 10 insertions, 13 deletions
diff --git a/ethosu/vela/high_level_command_to_npu_op.py b/ethosu/vela/high_level_command_to_npu_op.py index f7864442..efd8a03d 100644 --- a/ethosu/vela/high_level_command_to_npu_op.py +++ b/ethosu/vela/high_level_command_to_npu_op.py @@ -171,20 +171,17 @@ def create_padding(cmd: NpuStripe, primary_op: Operation) -> NpuPadding: def get_region(tens: Tensor, arch: ArchitectureFeatures) -> int: - if arch.feature_map_storage_mem_area == arch.fast_storage_mem_area: - base_ptr_idx_map = { - MemType.Permanent_NPU: BasePointerIndex.WeightTensor, - MemType.Permanent_CPU: BasePointerIndex.WeightTensor, - MemType.Scratch: BasePointerIndex.ScratchTensor, - MemType.Scratch_fast: BasePointerIndex.ScratchTensor, - } + base_ptr_idx_map = { + MemType.Permanent_NPU: BasePointerIndex.WeightTensor, + MemType.Permanent_CPU: BasePointerIndex.WeightTensor, + MemType.Scratch: BasePointerIndex.ScratchTensor, + } + + if arch.is_spilling_enabled(): + base_ptr_idx_map[MemType.Scratch_fast] = BasePointerIndex.ScratchFastTensor else: - base_ptr_idx_map = { - MemType.Permanent_NPU: BasePointerIndex.WeightTensor, - MemType.Permanent_CPU: BasePointerIndex.WeightTensor, - MemType.Scratch: BasePointerIndex.ScratchTensor, - MemType.Scratch_fast: BasePointerIndex.ScratchFastTensor, - } + base_ptr_idx_map[MemType.Scratch_fast] = BasePointerIndex.ScratchTensor + return int(base_ptr_idx_map[tens.mem_type]) |