aboutsummaryrefslogtreecommitdiff
path: root/ethosu/vela/register_command_stream_generator.py
diff options
context:
space:
mode:
authorLouis Verhaard <louis.verhaard@arm.com>2020-11-18 13:35:06 +0100
committerLouis Verhaard <louis.verhaard@arm.com>2020-11-23 13:39:56 +0100
commit5207830e53d454793cebfffa96d366ded79d71da (patch)
treeb99da4124607c6d20ceef9605eccf89e150ba666 /ethosu/vela/register_command_stream_generator.py
parentaeae56770f3c19182d32cc63fd32396e061a7648 (diff)
downloadethos-u-vela-5207830e53d454793cebfffa96d366ded79d71da.tar.gz
MLBEDSW-3425: Added external API for driver actions
Added external API to add driver actions to a command stream. Change-Id: Ie4779c1c745defc5769fa694358470cd6aea191c Signed-off-by: Louis Verhaard <louis.verhaard@arm.com>
Diffstat (limited to 'ethosu/vela/register_command_stream_generator.py')
-rw-r--r--ethosu/vela/register_command_stream_generator.py13
1 files changed, 2 insertions, 11 deletions
diff --git a/ethosu/vela/register_command_stream_generator.py b/ethosu/vela/register_command_stream_generator.py
index 04f7072d..9d79d58a 100644
--- a/ethosu/vela/register_command_stream_generator.py
+++ b/ethosu/vela/register_command_stream_generator.py
@@ -55,6 +55,7 @@ from .api import NpuTileBox
from .architecture_features import Accelerator
from .architecture_features import ArchitectureFeatures
from .architecture_features import Block
+from .architecture_features import create_default_arch
from .architecture_features import Rect
from .architecture_features import SharedBufferArea
from .architecture_features import SHRAMElements
@@ -1282,16 +1283,6 @@ def generate_register_command_stream(npu_op_list: List[NpuOperation], npu_accele
"""
accelerator = Accelerator.from_npu_accelerator(npu_accelerator)
emit = CommandStreamEmitter()
- arch = ArchitectureFeatures(
- vela_config_files=None,
- accelerator_config=accelerator.value,
- system_config=ArchitectureFeatures.DEFAULT_CONFIG,
- memory_mode=ArchitectureFeatures.DEFAULT_CONFIG,
- override_block_config=None,
- block_config_limit=None,
- max_blockdep=ArchitectureFeatures.MAX_BLOCKDEP,
- weight_estimation_scaling=1.0,
- verbose_config=False,
- )
+ arch = create_default_arch(accelerator)
generate_command_stream(emit, npu_op_list, arch)
return emit.to_list()