commit 03d40a24de88452a877f356ffe20e93705e76c43
author Tim Hall <tim.hall@arm.com> Thu Apr 22 12:08:28 2021 +0100
committer tim.hall <tim.hall@arm.com> Fri May 07 08:51:39 2021 +0000
tree e79c3a38df2617fc4d93a6849ed7b011710df5a6
parent f5c07c45b48717fb6391adb35cb73ca7cd5734c3

vela: Remove unused serialisation params

 - Removed unused nng parameter

Signed-off-by: Tim Hall <tim.hall@arm.com>
Change-Id: I0bb2eb101a84ea8022c8eb7bcbd86d617e933510

ethosu/vela/compiler_driver.py

diff --git a/ethosu/vela/compiler_driver.py b/ethosu/vela/compiler_driver.py
index 26d350e..e2c71ac 100644
--- a/ethosu/vela/compiler_driver.py
+++ b/ethosu/vela/compiler_driver.py
@@ -292,10 +292,10 @@
             nng, sg, arch, options.verbose_register_command_stream
         )
         scratch_tens, scratch_fast_tens, flash_tens = npu_serialisation.serialise_npu_subgraph_into_tensors(
-            nng, sg, arch, scratch_tens, scratch_fast_tens, flash_tens
+            sg, arch, scratch_tens, scratch_fast_tens, flash_tens
         )
 
-    npu_serialisation.rewrite_npu_call_ops(nng, root_sg, arch)
+    npu_serialisation.rewrite_npu_call_ops(root_sg, arch)
 
     # Set Scratch and Fast_scratch Tensor size
     if scratch_tens is not None:

ethosu/vela/npu_serialisation.py

diff --git a/ethosu/vela/npu_serialisation.py b/ethosu/vela/npu_serialisation.py
index 27c46ea..ad4d29c 100644
--- a/ethosu/vela/npu_serialisation.py
+++ b/ethosu/vela/npu_serialisation.py
@@ -57,7 +57,7 @@
     memory_tensor.values[start_addr:end_addr] = values
 
 
-def serialise_npu_subgraph_into_tensors(nng, sg, arch, scratch_tens, scratch_fast_tens, flash_tens):
+def serialise_npu_subgraph_into_tensors(sg, arch, scratch_tens, scratch_fast_tens, flash_tens):
     if sg.placement != PassPlacement.Npu:
         return scratch_tens, scratch_fast_tens, flash_tens
 
@@ -135,7 +135,7 @@
     startup_cps.outputs.insert(0, tens)
 
 
-def rewrite_npu_call_ops(nng, sg, arch):
+def rewrite_npu_call_ops(sg, arch):
     if sg.placement != PassPlacement.Cpu:
         return