diff options
-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 |