diff options
Diffstat (limited to 'ethosu/vela/compiler_driver.py')
| -rw-r--r-- | ethosu/vela/compiler_driver.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/ethosu/vela/compiler_driver.py b/ethosu/vela/compiler_driver.py index 64aff06b..b6a98a64 100644 --- a/ethosu/vela/compiler_driver.py +++ b/ethosu/vela/compiler_driver.py @@ -144,13 +144,14 @@ def compiler_driver(nng, arch, options, scheduler_options): # processed first during serialization into tensors first_npu_sg = nng.subgraphs[1] assert first_npu_sg.placement == PassPlacement.Npu + # Use the linear allocator for constant tensors tensor_allocation.allocate_tensors( nng, first_npu_sg, arch, permanent_storage, scheduler_options.use_ifm_ofm_overlap, - options.tensor_allocator, + TensorAllocator.LinearAlloc, options.verbose_allocation, options.show_minimum_possible_allocation, lr_graph_flash, @@ -195,7 +196,7 @@ def compiler_driver(nng, arch, options, scheduler_options): arch, permanent_storage, scheduler_options.use_ifm_ofm_overlap, - options.tensor_allocator, + TensorAllocator.LinearAlloc, options.verbose_allocation, options.show_minimum_possible_allocation, ) |