diff options
Diffstat (limited to 'ethosu/vela/compiler_driver.py')
| -rw-r--r-- | ethosu/vela/compiler_driver.py | 4 |
1 files changed, 2 insertions, 2 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: |