diff options
| author | Jacob Bohlin <jacob.bohlin@arm.com> | 2021-08-09 12:22:51 +0100 |
|---|---|---|
| committer | Jacob Bohlin <jacob.bohlin@arm.com> | 2021-08-16 17:40:20 +0200 |
| commit | b8060f51dcd6c128629e34855fdf46865ec165f5 (patch) | |
| tree | 64cf72fd162115dd15d16899fbcfe04fdb655c8f /ethosu/vela/register_command_stream_generator.py | |
| parent | 8f78ac2ff735b7c0be7787d6423eb96a0d8b5983 (diff) | |
| download | ethos-u-vela-b8060f51dcd6c128629e34855fdf46865ec165f5.tar.gz | |
MLBEDSW-4738 Reinstate find_block_config from v2.1
Reinstated the v2.1.0 functionality for find_block_configs(). This is
used exclusively by the external API.
Signed-off-by: Jacob Bohlin <jacob.bohlin@arm.com>
Change-Id: I6977f13866957edb083769658cc8c57c2b3556fb
Diffstat (limited to 'ethosu/vela/register_command_stream_generator.py')
| -rw-r--r-- | ethosu/vela/register_command_stream_generator.py | 17 |
1 files changed, 0 insertions, 17 deletions
diff --git a/ethosu/vela/register_command_stream_generator.py b/ethosu/vela/register_command_stream_generator.py index b4a633e9..d61e5717 100644 --- a/ethosu/vela/register_command_stream_generator.py +++ b/ethosu/vela/register_command_stream_generator.py @@ -1025,23 +1025,6 @@ def generate_command_stream( return res -# ------------------------------------------------------------------- -# EXTERNAL API -# ------------------------------------------------------------------- - - -def find_block_configs(npu_op: NpuOperation, npu_accelerator: NpuAccelerator) -> List[NpuShape3D]: - """ - Internal implementation of the public facing API for finding block configs. - """ - if isinstance(npu_op, NpuBlockOperation): - # TODO: implement this function - arch = create_default_arch(Accelerator.from_npu_accelerator(npu_accelerator)) - block = arch.ofm_ublock - return [NpuShape3D(height=block.height, width=block.width, depth=block.depth)] - return [] - - def generate_register_command_stream(npu_op_list: List[NpuOperation], npu_accelerator: NpuAccelerator) -> List[int]: """ Internal implementation of the public facing API for generating an Ethos-U register command stream. |