aboutsummaryrefslogtreecommitdiff
path: root/ethosu/vela/insert_dma.py
diff options
context:
space:
mode:
Diffstat (limited to 'ethosu/vela/insert_dma.py')
-rw-r--r--ethosu/vela/insert_dma.py8
1 files changed, 1 insertions, 7 deletions
diff --git a/ethosu/vela/insert_dma.py b/ethosu/vela/insert_dma.py
index 6c5c8031..6cd2202c 100644
--- a/ethosu/vela/insert_dma.py
+++ b/ethosu/vela/insert_dma.py
@@ -61,13 +61,7 @@ def insert_dma_cmd(op, arch):
dma_cmd.attrs["destination"] = new_tens.mem_area
dma_cmd.run_on_npu = True
if tens.purpose == TensorPurpose.LUT:
- # TODO: Add support more than one LUT at a time
- # Reserve last 2 blocks for LUT
- if arch.shram_reserved_unused_banks == 0:
- arch.shram_reserved_unused_banks = 2
- arch.shram_total_banks -= arch.shram_reserved_unused_banks
- # Place the LUT in the last 2 blocks of SHRAM
- new_tens.address = arch.shram_bank_size * arch.shram_total_banks
+ new_tens.mem_area = MemArea.Shram
op.inputs[idx] = new_tens
return op