From 5207830e53d454793cebfffa96d366ded79d71da Mon Sep 17 00:00:00 2001 From: Louis Verhaard Date: Wed, 18 Nov 2020 13:35:06 +0100 Subject: 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 --- ethosu/vela/npu_serialisation.py | 14 +------------- 1 file changed, 1 insertion(+), 13 deletions(-) (limited to 'ethosu/vela/npu_serialisation.py') diff --git a/ethosu/vela/npu_serialisation.py b/ethosu/vela/npu_serialisation.py index 04534cce..a11907b2 100644 --- a/ethosu/vela/npu_serialisation.py +++ b/ethosu/vela/npu_serialisation.py @@ -15,8 +15,6 @@ # limitations under the License. # Description: # Serialises and packs an NPU subgraph into tensors. -import struct - import numpy as np from . import driver_actions @@ -70,17 +68,7 @@ def serialise_npu_subgraph_into_tensors(nng, sg, arch, scratch_tens, scratch_fas flash_size = sg.memory_used.get(flash_area, 0) scratch_size = sg.memory_used.get(scratch_area, 0) - # Prepare driver actions for this command tensor - da_list = [] - driver_actions.emit_fourcc(da_list, "COP1") - driver_actions.emit_config(da_list, 0, 1, arch) - driver_actions.emit_cmd_stream_header(da_list, len(sg.register_command_stream)) - - # Append command stream words - da_list.extend(sg.register_command_stream) - - # Convert to bytes - payload_bytes = struct.pack("<{0}I".format(len(da_list)), *da_list) + payload_bytes = driver_actions.create_driver_payload(sg.register_command_stream, arch) command_stream_size_bytes = len(payload_bytes) -- cgit v1.2.1