diff options
| author | Michael McGeagh <michael.mcgeagh@arm.com> | 2020-08-07 11:54:28 +0100 |
|---|---|---|
| committer | Fredrik Knutsson <fredrik.knutsson.hunnebo@gmail.com> | 2020-08-12 06:30:46 +0000 |
| commit | c5b549b599ff459a29115a48e8f067eaa5891638 (patch) | |
| tree | 62b22dfda83f4665b10f549d9f9d7ddaedff15a1 /ethosu/vela/insert_dma.py | |
| parent | 5778ffdab61a46369c73c91f2c6289ba9833e3a3 (diff) | |
| download | ethos-u-vela-c5b549b599ff459a29115a48e8f067eaa5891638.tar.gz | |
MLBEDSW-2637 Utilise new tensor and operator funcs
add_input_tensor, set_output_tensor, create_const_tensor and
create_reshape_tensor have recently been added.
This replaces all found existing instances with these new helper
functions
Signed-off-by: Michael McGeagh <michael.mcgeagh@arm.com>
Change-Id: If33be8dbf237b2087b562b03cdeb51da1f99a786
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 |