diff options
| author | Tim Hall <tim.hall@arm.com> | 2021-04-22 12:08:28 +0100 |
|---|---|---|
| committer | tim.hall <tim.hall@arm.com> | 2021-05-07 08:51:39 +0000 |
| commit | 03d40a24de88452a877f356ffe20e93705e76c43 (patch) | |
| tree | e79c3a38df2617fc4d93a6849ed7b011710df5a6 | |
| parent | f5c07c45b48717fb6391adb35cb73ca7cd5734c3 (diff) | |
| download | ethos-u-vela-03d40a24de88452a877f356ffe20e93705e76c43.tar.gz | |
vela: Remove unused serialisation params
- Removed unused nng parameter
Signed-off-by: Tim Hall <tim.hall@arm.com>
Change-Id: I0bb2eb101a84ea8022c8eb7bcbd86d617e933510
| -rw-r--r-- | ethosu/vela/compiler_driver.py | 4 | ||||
| -rw-r--r-- | ethosu/vela/npu_serialisation.py | 4 |
2 files changed, 4 insertions, 4 deletions
diff --git a/ethosu/vela/compiler_driver.py b/ethosu/vela/compiler_driver.py index 26d350ea..e2c71ac0 100644 --- a/ethosu/vela/compiler_driver.py +++ b/ethosu/vela/compiler_driver.py @@ -292,10 +292,10 @@ def compiler_driver(nng, arch, options, scheduler_options): 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: diff --git a/ethosu/vela/npu_serialisation.py b/ethosu/vela/npu_serialisation.py index 27c46ea0..ad4d29ca 100644 --- a/ethosu/vela/npu_serialisation.py +++ b/ethosu/vela/npu_serialisation.py @@ -57,7 +57,7 @@ def copy_ifm_values_to_memory_tensor(memory_tensor, src_tensor): 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 @@ def add_const_tens_to_startup_cascaded_pass(startup_cps, tens): 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 |