aboutsummaryrefslogtreecommitdiff
path: root/ethosu/vela/npu_serialisation.py
diff options
context:
space:
mode:
authorerik.andersson@arm.com <erik.andersson@arm.com>2021-03-16 09:40:24 +0100
committererik.andersson@arm.com <erik.andersson@arm.com>2021-03-19 14:05:17 +0100
commit1878dab5f2fb860ae98e5e1dafcdb5cec7d33349 (patch)
tree55e1c5c076453bf8df8572bc10c073d94782b5b9 /ethosu/vela/npu_serialisation.py
parenta8e48e6792216974d41c7fc69d9406916c153fba (diff)
downloadethos-u-vela-1878dab5f2fb860ae98e5e1dafcdb5cec7d33349.tar.gz
MLBEDSW-3458: Added command stream size check.
If the command stream size exceeds a certain threshold, a VelaError will now be raised. Signed-off-by: erik.andersson@arm.com <erik.andersson@arm.com> Change-Id: I9b9383f4c298a778b160cd527374e9244e4cae26
Diffstat (limited to 'ethosu/vela/npu_serialisation.py')
-rw-r--r--ethosu/vela/npu_serialisation.py1
1 files changed, 0 insertions, 1 deletions
diff --git a/ethosu/vela/npu_serialisation.py b/ethosu/vela/npu_serialisation.py
index 66230537..27c46ea0 100644
--- a/ethosu/vela/npu_serialisation.py
+++ b/ethosu/vela/npu_serialisation.py
@@ -119,7 +119,6 @@ def serialise_npu_subgraph_into_tensors(nng, sg, arch, scratch_tens, scratch_fas
ps.ifm2_tensor.mem_type not in (MemType.Scratch, MemType.Scratch_fast)
):
copy_ifm_values_to_memory_tensor(sg.flash_tensor, ps.ifm2_tensor)
-
sg.command_stream_tensor = make_memory_tensor(
sg.name + "_command_stream", flash_area, MemType.Permanent_CPU, command_stream_size_bytes, True, arch
)