commit 0b9c9a3873da3d368e184308f4f9a4c202e3fb67
author Louis Verhaard <louis.verhaard@arm.com> Tue Sep 15 14:05:38 2020 +0200
committer Louis Verhaard <louis.verhaard@arm.com> Fri Sep 25 08:32:55 2020 +0200
tree f057bfeacd6c6323cd0e7f33cd7ec33f941b4f8d
parent 8854dc9088586e1bb0bf2640b2289903dfa3c822

MLBEDSW-2337: Intermediate feature maps in fast storage

Attempts to use fast storage for feature maps used in between
cascaded passes.

This is only relevant for system configurations where feature maps
are by default not placed in SRAM, but there is SRAM for fast storage.

Change-Id: I207b7cf32cfcb5bea3e6b93c2da1161c4af5221d
Signed-off-by: Louis Verhaard <louis.verhaard@arm.com>

ethosu/vela/tensor.py

diff --git a/ethosu/vela/tensor.py b/ethosu/vela/tensor.py
index eedbada..c0786bf 100644
--- a/ethosu/vela/tensor.py
+++ b/ethosu/vela/tensor.py
@@ -278,7 +278,7 @@
     def set_address_for_tens(cls, tens_id, mem_type, address):
         # Check previous address if there is one
         previous_address = cls.address_map[tens_id].get(mem_type)
-        if previous_address is not None:
+        if address is not None and previous_address is not None:
             assert previous_address == address, "Two different addresses cannot be assigned to the same tensor."
 
         # Set tensor's address for memory type