diff options
Diffstat (limited to 'ethosu/vela/insert_dma.py')
-rw-r--r-- | ethosu/vela/insert_dma.py | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/ethosu/vela/insert_dma.py b/ethosu/vela/insert_dma.py index 76016f1f..6c5c8031 100644 --- a/ethosu/vela/insert_dma.py +++ b/ethosu/vela/insert_dma.py @@ -56,11 +56,10 @@ def insert_dma_cmd(op, arch): new_tens = tens.clone_into_fast_storage(arch) dma_cmd = Operation("DMA", tens.ops[0].name + "_dma") dma_cmd.inputs = [tens] - dma_cmd.outputs = [new_tens] + dma_cmd.set_output_tensor(new_tens) dma_cmd.attrs["source"] = tens.mem_area dma_cmd.attrs["destination"] = new_tens.mem_area dma_cmd.run_on_npu = True - new_tens.ops = [dma_cmd] if tens.purpose == TensorPurpose.LUT: # TODO: Add support more than one LUT at a time # Reserve last 2 blocks for LUT |